Move to Ubuntu 20.04 for cuda 10.2 toolkit image (NVIDIA#3477)

Signed-off-by: Janusz Lisiecki <[email protected]>
AbhijitManepatil · Nov 9, 2021 · f3d8522 · f3d8522
1 parent 93ff988
commit f3d8522
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/docker/Dockerfile.cuda102.x86_64.deps b/docker/Dockerfile.cuda102.x86_64.deps
@@ -1,12 +1,15 @@
-ARG TOOLKIT_BASE_IMAGE=ubuntu:18.04
+ARG TOOLKIT_BASE_IMAGE=ubuntu:20.04
 FROM ${TOOLKIT_BASE_IMAGE} as cuda
 
+ENV DEBIAN_FRONTEND=noninteractive
+
 RUN apt update && apt install -y libxml2 curl perl gcc && \
     rm -rf /var/lib/apt/lists/*
 
 RUN curl -LO https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run && \
     chmod +x cuda_*.run && \
-    ./cuda_*.run --silent --no-opengl-libs --toolkit && \
+    # just installing with override on Ubuntu 20:04 doesn't work for some reason so unpack and move
+    ./cuda_*.run --extract=/tmp/cuda/ --override && mv /tmp/cuda/cuda-toolkit/ /usr/local/cuda && \
     rm -f cuda_*.run;
 
 FROM scratch