cpu and gpu

Author: cmanzo

docker/Dockerfile

@@ -33,18 +33,14 @@ RUN pip install --no-cache-dir \
 COPY ./requirements.txt /tmp/
 RUN pip install --no-cache-dir -r /tmp/requirements.txt
 
-# Install all widget bits via pip
-RUN pip install --no-cache-dir \
-      ipywidgets \
-      jupyterlab_widgets \
-      widgetsnbextension
-
-# Enable classic-notebook widget nbextension
-RUN jupyter nbextension install --py widgetsnbextension --sys-prefix && \
-    jupyter nbextension enable  --py widgetsnbextension --sys-prefix
-
-# Enable the server extension for JupyterLab & VS Code
-RUN jupyter server extension enable --py jupyterlab_widgets --sys-prefix
+# Enable Jupyter widget support
+RUN pip install --no-cache-dir ipywidgets
+
+# For JupyterLab (inside the container), install the widget manager
+RUN jupyter labextension install @jupyter-widgets/jupyterlab-manager \
+  --no-build && \
+  jupyter lab build && \
+  jupyter labextension enable @jupyter-widgets/jupyterlab-manager
 
 EXPOSE 8888
 CMD ["start-notebook.sh"]

docker/Dockerfile.gpu

@@ -29,18 +29,14 @@ RUN pip install --no-cache-dir \
 COPY ./requirements.txt /tmp/
 RUN pip install --no-cache-dir -r /tmp/requirements.txt
 
-# Install all widget bits via pip
-RUN pip install --no-cache-dir \
-      ipywidgets \
-      jupyterlab_widgets \
-      widgetsnbextension
-
-# Enable classic-notebook widget nbextension
-RUN jupyter nbextension install --py widgetsnbextension --sys-prefix && \
-    jupyter nbextension enable  --py widgetsnbextension --sys-prefix
-
-# Enable the server extension for JupyterLab & VS Code
-RUN jupyter server extension enable --py jupyterlab_widgets --sys-prefix
+# Enable Jupyter widget support
+RUN pip install --no-cache-dir ipywidgets
+
+# For JupyterLab (inside the container), install the widget manager
+RUN jupyter labextension install @jupyter-widgets/jupyterlab-manager \
+  --no-build && \
+  jupyter lab build && \
+  jupyter labextension enable @jupyter-widgets/jupyterlab-manager
 
 EXPOSE 8888
 CMD ["start-notebook.sh"]