cpu and gpu

Author: cmanzo

docker/Dockerfile

@@ -14,9 +14,9 @@ WORKDIR /home/jovyan/work
 RUN git clone --depth 1 https://github.com/DeepTrackAI/DeepLearningCrashCourse.git .
 
 # ---------- ensure correct ownership ---------------------------------------------
-    USER root
-    RUN chown -R ${NB_UID}:${NB_GID} /home/jovyan/work
-    USER ${NB_UID}
+USER root
+RUN chown -R ${NB_UID}:${NB_GID} /home/jovyan/work
+USER ${NB_UID}
 
 # ---------- extra Python deps -----------------------------------------------------
 # CPU-only PyTorch (wheels from CPU index)
@@ -29,6 +29,7 @@ RUN pip install --no-cache-dir \
         torch-geometric \
         -f https://data.pyg.org/whl/torch-2.7.0+cpu.html
 
+# Then install your extra deps
 COPY ./requirements.txt /tmp/
 RUN pip install --no-cache-dir -r /tmp/requirements.txt
 

docker/Dockerfile.gpu

@@ -2,21 +2,23 @@
 # GPU-enabled Jupyter + PyTorch + your course
 # ────────────────────────────────────────────────────────────────────────────────
 FROM quay.io/jupyter/pytorch-notebook:cuda12-python-3.11.9
-
+# ---------- system utilities ------------------------------------------------------
 USER root
 RUN apt-get update && \
     apt-get install -y --no-install-recommends git && \
     rm -rf /var/lib/apt/lists/*
 USER ${NB_UID}
 
+# ---------- copy the course -------------------------------------------------------
 WORKDIR /home/jovyan/work
 RUN git clone --depth 1 https://github.com/DeepTrackAI/DeepLearningCrashCourse.git .
 
-# ensure jovyan owns the workspace
+# ---------- ensure correct ownership ---------------------------------------------
 USER root
 RUN chown -R ${NB_UID}:${NB_GID} /home/jovyan/work
 USER ${NB_UID}
 
+# ---------- extra Python deps -----------------------------------------------------
 # Base image already has CUDA-enabled PyTorch 2.7+
 # Install PyG CUDA wheels matching torch 2.7.0+cu121
 RUN pip install --no-cache-dir \
@@ -36,6 +38,5 @@ RUN pip install --no-cache-dir \
 RUN jupyter nbextension install --py widgetsnbextension --sys-prefix && \
     jupyter nbextension enable  --py widgetsnbextension --sys-prefix
 
-
 EXPOSE 8888
 CMD ["start-notebook.sh"]