cpu and gpu

Author: cmanzo

docker/Dockerfile

@@ -11,7 +11,6 @@ USER ${NB_UID}
 
 # ---------- copy the course -------------------------------------------------------
 WORKDIR /home/jovyan/work
-# COPY .. .
 RUN git clone --depth 1 https://github.com/DeepTrackAI/DeepLearningCrashCourse.git .
 
 # ---------- ensure correct ownership ---------------------------------------------

docker/Dockerfile.gpu

@@ -1,4 +1,3 @@
-# docker/Dockerfile.gpu
 # ────────────────────────────────────────────────────────────────────────────────
 # GPU-enabled Jupyter + PyTorch + your course
 # ────────────────────────────────────────────────────────────────────────────────
@@ -34,9 +33,7 @@ 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
+  jupyter lab build
 
 EXPOSE 8888
 CMD ["start-notebook.sh"]