cpu and gpu

Author: cmanzo

docker/Dockerfile

@@ -33,10 +33,18 @@ RUN pip install --no-cache-dir \
 COPY ./requirements.txt /tmp/
 RUN pip install --no-cache-dir -r /tmp/requirements.txt
 
-# Install widget support
+# Install all widget bits via pip
 RUN pip install --no-cache-dir \
       ipywidgets \
-      jupyterlab_widgets
+      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
 
 EXPOSE 8888
 CMD ["start-notebook.sh"]

docker/Dockerfile.gpu

@@ -29,10 +29,18 @@ RUN pip install --no-cache-dir \
 COPY ./requirements.txt /tmp/
 RUN pip install --no-cache-dir -r /tmp/requirements.txt
 
-# Install widget support
+# Install all widget bits via pip
 RUN pip install --no-cache-dir \
       ipywidgets \
-      jupyterlab_widgets
+      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
 
 EXPOSE 8888
 CMD ["start-notebook.sh"]