diff --git a/Dockerfile b/Dockerfile index 5b389cf..dbbf2ac 100644 --- a/Dockerfile +++ b/Dockerfile @@ -29,7 +29,10 @@ RUN apt-get update && \ 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/ /" && \ + add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /" && \ + rm -rf /var/lib/apt/lists/* && \ + rm -rf /etc/apt/sources.list.d/cuda.list && \ + rm -rf /etc/apt/sources.list.d/nvidia-ml.list && \ apt-get update && \ apt-get -y install cuda && \ # 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 && \