fix

Author: KevinZonda

MLBook.Dockerfile

@@ -3,14 +3,15 @@ FROM jupyterhub/singleuser:latest
 WORKDIR /tmp
 
 RUN mamba install -y \
-    pytorch \
+    pytorch torchvision torchaudio pytorch-cuda \
     tensorflow \
     keras \
     numpy \
     seaborn \
     pandas \
     matplotlib \
-    scikit-learn
+    scikit-learn \
+    -c pytorch -c nvidia
 
 RUN mamba clean --all -f -y && \
     jupyter lab clean && \
@@ -19,4 +20,5 @@ RUN mamba clean --all -f -y && \
     fix-permissions "/home/${NB_USER}"
 
 USER ${NB_UID}
-WORKDIR "${HOME}"
+WORKDIR "${HOME}"
+COPY pip.conf /etc/pip.conf

TorchBook.Dockerfile

@@ -3,12 +3,13 @@ FROM jupyterhub/singleuser:latest
 WORKDIR /tmp
 
 RUN mamba install -y \
-    pytorch \
+    pytorch torchvision torchaudio pytorch-cuda \
     numpy \
     seaborn \
     pandas \
     matplotlib \
-    scikit-learn
+    scikit-learn \
+    -c pytorch -c nvidia
 
 RUN mamba clean --all -f -y && \
     jupyter lab clean && \
@@ -17,4 +18,5 @@ RUN mamba clean --all -f -y && \
     fix-permissions "/home/${NB_USER}"
 
 USER ${NB_UID}
-WORKDIR "${HOME}"
+WORKDIR "${HOME}"
+COPY pip.conf /etc/pip.conf