Skip to content

Commit

Permalink
Update Dockerfile
Browse files Browse the repository at this point in the history
Simplify back to previous stable build + added path from 2022 version.
  • Loading branch information
uebergeek999 committed Nov 16, 2024
1 parent 69c6f20 commit 36fd2e6
Showing 1 changed file with 6 additions and 27 deletions.
33 changes: 6 additions & 27 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -6,26 +6,8 @@ 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/*

# 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 && \
Expand All @@ -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
Expand All @@ -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

Expand Down

0 comments on commit 36fd2e6

Please sign in to comment.