diff --git a/Dockerfile b/Dockerfile index 6b2b3fa..8592e8e 100644 --- a/Dockerfile +++ b/Dockerfile @@ -6,26 +6,8 @@ LABEL maintainer="UC San Diego ITS/ETS " 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/* -# 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 +# install cuda toolkit 11-1 # 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 RUN apt-get update && \ apt-get install -y p7zip-full software-properties-common && \ @@ -39,13 +21,10 @@ 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 && \ - - 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 && \ - dpkg -i /var/tmp/libcudnn8.deb /var/tmp/libnccl2.deb && \ - fix-permissions $CONDA_DIR && \ - fix-permissions /home/$NB_USER + apt-get install -qqy cuda-11-1 cuda-nvcc-11-1 cuda-toolkit-11-1 && \ + + fix-permissions $CONDA_DIR && \ + fix-permissions /home/$NB_USER COPY env.yml /tmp/env.yml @@ -64,7 +43,7 @@ RUN conda env create --file /tmp/env.yml && \ USER jovyan # other packages here... - +# From the "original" 2022 version: USER $NB_UID:$NB_GID ENV PATH=${PATH}:/usr/local/cuda/bin