cpu and gpu

Author: cmanzo

docker/Dockerfile

@@ -30,9 +30,16 @@ RUN pip install --no-cache-dir \
         torch-geometric \
         -f https://data.pyg.org/whl/torch-2.7.0+cpu.html
 
-
 COPY ./requirements.txt /tmp/
 RUN pip install --no-cache-dir -r /tmp/requirements.txt
 
+# 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
+
 EXPOSE 8888
 CMD ["start-notebook.sh"]

docker/Dockerfile.gpu

@@ -28,5 +28,15 @@ RUN pip install --no-cache-dir \
 COPY ./requirements.txt /tmp/
 RUN pip install --no-cache-dir -r /tmp/requirements.txt
 
+# 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 && \
+    jupyter labextension install @jupyter-widgets/jupyterlab-manager --no-build
+
 EXPOSE 8888
 CMD ["start-notebook.sh"]