From 8440ec5e8cb3eba619775f9d40fd94bfc6e1ef44 Mon Sep 17 00:00:00 2001 From: Faraz Fallahi Date: Sun, 20 Jun 2021 02:09:39 -0400 Subject: [PATCH] Update Dockerfile --- Dockerfile | 26 +++++++++++++++----------- 1 file changed, 15 insertions(+), 11 deletions(-) diff --git a/Dockerfile b/Dockerfile index 16b8caa..b6a9096 100644 --- a/Dockerfile +++ b/Dockerfile @@ -36,6 +36,21 @@ RUN \ bazel --version && \ exit 0 +# Install cuda +RUN \ + set -eux && \ + wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin && \ + mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600 && \ + apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub && \ + add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /" && \ + apt-get update && \ + apt-get -y install cuda && \ + apt-get -y autoremove && \ + apt-get -y autoclean && \ + apt-get -y clean && \ + rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/* && \ + exit 0 + # Install python pip RUN \ set -eux && \ @@ -107,17 +122,6 @@ RUN \ ssh-keygen -A && \ exit 0 -# Install cuda -RUN \ - set -eux && \ - wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin && \ - mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600 && \ - apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub && \ - add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /" && \ - apt-get update && \ - apt-get -y install cuda && \ - exit 0 - ENV IGNORE_CC_MISMATCH=1 ENV PATH=$PATH:/usr/local/cuda/bin ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64