From e888732d081de08cbd49585dd7a5a00817cb672f Mon Sep 17 00:00:00 2001 From: Shaquille Soekhlal Date: Tue, 2 Feb 2021 11:45:34 +0100 Subject: [PATCH] updated cuda --- .build/Dockerfile | 2 +- src/Dockerfile.gpulibs | 2 +- src/Dockerfile.header | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.build/Dockerfile b/.build/Dockerfile index 1e26f9f..fbed694 100755 --- a/.build/Dockerfile +++ b/.build/Dockerfile @@ -7,7 +7,7 @@ # Use NVIDIA CUDA as base image and run the same installation as in the other packages. # The version of cudatoolkit must match those of the base image, see Dockerfile.pytorch -FROM nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04 +FROM nvidia/cuda:11.2.0-runtime-ubuntu20.04 LABEL authors="Christoph Schranz , Mathematical Michael " # This is a concatenated Dockerfile, the maintainers of subsequent sections may vary. RUN chmod 1777 /tmp && chmod 1777 /var/tmp diff --git a/src/Dockerfile.gpulibs b/src/Dockerfile.gpulibs index e88157e..2c60c5b 100644 --- a/src/Dockerfile.gpulibs +++ b/src/Dockerfile.gpulibs @@ -14,7 +14,7 @@ RUN conda install --quiet --yes \ RUN conda install --quiet --yes \ pytorch \ torchvision \ - cudatoolkit=10.1 -c pytorch + cudatoolkit=11.2 -c pytorch # pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cu90/torch_nightly.html && \ RUN pip install --no-cache-dir torchviz diff --git a/src/Dockerfile.header b/src/Dockerfile.header index 7453ceb..f72e93a 100644 --- a/src/Dockerfile.header +++ b/src/Dockerfile.header @@ -1,6 +1,6 @@ # Use NVIDIA CUDA as base image and run the same installation as in the other packages. # The version of cudatoolkit must match those of the base image, see Dockerfile.pytorch -FROM nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04 +FROM nvidia/cuda:11.2.0-runtime-ubuntu20.04 LABEL authors="Christoph Schranz , Mathematical Michael " # This is a concatenated Dockerfile, the maintainers of subsequent sections may vary. RUN chmod 1777 /tmp && chmod 1777 /var/tmp