#1469: docker: add support for nvidia nvcc 10.2 (useful in the future)

DARMA-tasking · Jun 18, 2021 · 4134874 · 4134874
1 parent a5387b4
commit 4134874
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/ci/docker/ubuntu-18.04-nvidia-cpp.dockerfile b/ci/docker/ubuntu-18.04-nvidia-cpp.dockerfile
@@ -42,7 +42,19 @@ RUN if test ${compiler} = "nvcc-10"; then \
       rm -rf /var/lib/apt/lists/* && \
       rm -rf cuda-repo-ubuntu1804-10-1-local-10.1.243-418.87.00_1.0-1_amd64.deb && \
       ln -s /usr/local/cuda-10.1 /usr/local/cuda-versioned; \
-    else \
+   elif test ${compiler} = "nvcc-10.2"; then \
+     wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin && \
+     mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600 && \
+     wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb && \
+     dpkg -i cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb && \
+     apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub && \
+     apt-get update && \
+     apt-get -y install cuda && \
+     apt-get clean && \
+     rm -rf /var/lib/apt/lists/* && \
+     rm -rf cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb && \
+     ln -s /usr/local/cuda-10.2 /usr/local/cuda-versioned; \
+   else \
       wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin && \
       mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600 && \
       wget http://developer.download.nvidia.com/compute/cuda/11.0.1/local_installers/cuda-repo-ubuntu1804-11-0-local_11.0.1-450.36.06-1_amd64.deb && \

diff --git a/docker-compose.yml b/docker-compose.yml
@@ -14,7 +14,7 @@
 #             clang-3.9, clang-4.0, clang-5.0, clang-6.0, clang-7, clang-8,
 #             clang-9, clang-10,
 #             icc-18, icc-19,
-#             nvcc-10, nvcc-11}
+#             nvcc-10, nvcc-10.2, nvcc-11}
 #   REPO=lifflander1/vt
 #   UBUNTU={18.04, 20.04}
 #   ULIMIT_CORE=0
@@ -87,6 +87,7 @@ volumes:
   amd64-ubuntu-18.04-icc-19-cache:
   amd64-ubuntu-18.04-icc-20-cache:
   amd64-ubuntu-18.04-nvcc-10-cache:
+  amd64-ubuntu-18.04-nvcc-10.2-cache:
   amd64-ubuntu-18.04-nvcc-11-cache:
   amd64-alpine-clang-3.9-cache:
   amd64-alpine-clang-4.0-cache:
@@ -106,6 +107,7 @@ volumes:
   amd64-alpine-icc-19-cache:
   amd64-alpine-icc-20-cache:
   amd64-alpine-nvcc-10-cache:
+  amd64-alpine-nvcc-10.2-cache:
   amd64-alpine-nvcc-11-cache:
   arm64v8-ubuntu-18.04-gcc-7-cache:
   arm64v8-alpine-gcc-7-cache: