Updated Dockerfile

Dockerfile

@@ -6,21 +6,24 @@ LABEL maintainer="UC San Diego ITS/ETS <[email protected]>"
 USER root
 
 
-RUN apt-get -q update && \
- apt-get -qy install apt-utils  && \
- apt-get -qy dist-upgrade && \
- apt-get -qy auto-remove && \
- apt-get install -qy p7zip-full software-properties-common && \
- apt-get clean && \
- rm -rf /var/lib/apt/lists/*  && \
-            https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
-RUN wget -nv https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin -O /etc/apt/preferences.d/cuda-repository-pin-600 && \
- apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub && \
- wget https://developer.download.nvidia.com/compute/cuda/repos/$distro/$arch/cuda-keyring_1.0-1_all.deb && \
- dpkg -i cuda-keyring_1.0-1_all.deb && \
- add-apt-repository "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /" && \
- apt-get -q update && \
- apt-get install -qqy cuda-11-1 cuda-nvcc-11-1	cuda-toolkit-11-1 && \
+RUN     apt-get -q update && \
+        apt-get -qy install apt-utils && \
+        apt-get -qy dist-upgrade && \
+        apt-get -qy auto-remove && \
+        apt-get install -qy \
+                p7zip-full \
+                software-properties-common && \
+        apt-get clean && \
+        rm -rf /var/lib/apt/lists/*
+
+# CUDA 11-1
+RUN     wget -nv https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin -O /etc/apt/preferences.d/cuda-repository-pin-600 && \
+        apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub && \
+        wget https://developer.download.nvidia.com/compute/cuda/repos/$distro/$arch/cuda-keyring_1.0-1_all.deb && \
+        dpkg -i cuda-keyring_1.0-1_all.deb && \
+        add-apt-repository "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /" && \
+        apt-get -q update && \
+        apt-get install -qqy cuda-11-1 cuda-nvcc-11-1	cuda-toolkit-11-1 && \
 
 # install cuda toolkit 
 # https://developer.nvidia.com/cuda-11.1.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=2004&target_type=debnetwork
@@ -36,7 +39,7 @@ RUN apt-get update && \
     rm -rf /etc/apt/sources.list.d/nvidia-ml.list && \
     apt-get update  && \
 #apt-get -y install cuda && \
-    apt-get install -qqy cuda-11-1 cuda-nvcc-11-1	cuda-toolkit-11-1 && \
+    apt-get install -qqy cuda-11-1 cuda-nvcc-11-1 cuda-toolkit-11-1 && \
 
   wget -nv https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcudnn8_8.0.5.39-1+cuda11.1_amd64.deb -O /var/tmp/libcudnn8.deb  && \
   wget -nv https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libnccl2_2.8.4-1+cuda11.1_amd64.deb -O /var/tmp/libnccl2.deb  && \
@@ -45,28 +48,6 @@ RUN apt-get update && \
   fix-permissions /home/$NB_USER
 
 
-# apt-get -q update
-# apt-get -qy install apt-utils
-# apt-get -qy dist-upgrade 
-# apt-get -qy auto-remove 
-# apt-get install -qy		p7zip-full 		software-properties-common 
-# apt-get clean 
-# rm -rf /var/lib/apt/lists/*"
-# wget -nv https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin -O /etc/apt/preferences.d/cuda-repository-pin-600 \u0026\u0026 	apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub \u0026\u0026 	add-apt-repository \"deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /\" \u0026\u0026 	apt-get -q update \u0026\u0026 	apt-get install -qqy 		cuda-11-1 		cuda-nvcc-11-1 		cuda-toolkit-11-1 \u0026\u0026 	wget -nv https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcudnn8_8.0.5.39-1+cuda11.1_amd64.deb -O /var/tmp/libcudnn8.deb \u0026\u0026 	wget -nv https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libnccl2_2.8.4-1+cuda11.1_amd64.deb -O /var/tmp/libnccl2.deb \u0026\u0026 	dpkg -i /var/tmp/libcudnn8.deb /var/tmp/libnccl2.deb"
-#fix-permissions $CONDA_DIR \u0026\u0026 	fix-permissions /home/$NB_USER"
-
-#/opt/conda/bin/python3 -m pip install --upgrade pip"
-#/opt/conda/bin/conda install -y 		jaxlib==0.1.55 		tensorboard"
-#pip install torch 		-f https://download.pytorch.org/whl/rocm4.0.1/torch_stable.html \u0026\u0026 	pip install ninja \u0026\u0026 	pip install 'git+https://github.com/pytorch/[email protected]'"
-#pip install --no-cache-dir 		torch==1.8.0+cu111 		torchvision==0.9.0+cu111 		torchaudio==0.8.0 		-f https://download.pytorch.org/whl/torch_stable.html"
-#pip install --no-cache-dir 		gdown 		imageio-ffmpeg==0.4.3 		jax==0.1.73 		opencv-contrib-python-headless 		opencv-python 		opensimplex 		pillow 		pyspng==0.1.0 		networkx 		scipy"
-#(nop)  USER root
-#(nop) COPY dir:36a36661fdff68aec2767c0def27d2808864e4eca0678ba1eb7e93151342e0cb in /usr/share/datahub/tests/scipy-ml-notebook "
-#chmod -R +x /usr/share/datahub/tests/scipy-ml-notebook \u0026\u0026 	chown -R 1000:100 /home/jovyan \u0026\u0026 	chmod +x /run_jupyter.sh"
-#(nop)  USER 1000:100
-#(nop)  ENV PATH=/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/cuda/bin
-
-
 COPY env.yml /tmp/env.yml
 
 RUN conda env create --file /tmp/env.yml && \