commit 474fe7d9763d9f5c5e318e669833041c682c019e
author Eli Uriegas <eliuriegas@fb.com> Fri Mar 05 14:41:43 2021 -0800
committer Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> Fri Mar 05 14:45:57 2021 -0800
tree 988ebdc8f863eff27577ba0f32c2b68ad9c87f32
parent f58f7b786ca6d3aa57fb916e30fa348f15318ab2

docker: Update default cuda => 11.1 (#53299)

Summary:
We no longer build binaries for CUDA 11.0 so let's ensure that we have
build for CUDA 11.1 by default instead

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/53299

Reviewed By: anjali411

Differential Revision: D26857194

Pulled By: seemethere

fbshipit-source-id: 6094913922c0da832b96e5e49a67369d69d0b8ad

Dockerfile

diff --git a/Dockerfile b/Dockerfile
index d4b2dcf..cea0fcb 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -5,7 +5,7 @@
 #
 #       If you do not use buildkit you are not going to have a good time
 #
-#       For reference: 
+#       For reference:
 #           https://docs.docker.com/develop/develop-images/build_enhancements/
 ARG BASE_IMAGE=ubuntu:18.04
 ARG PYTHON_VERSION=3.8
@@ -51,8 +51,8 @@
 
 FROM conda as conda-installs
 ARG PYTHON_VERSION=3.8
-ARG CUDA_VERSION=11.0
-ARG CUDA_CHANNEL=defaults
+ARG CUDA_VERSION=11.1
+ARG CUDA_CHANNEL=nvidia
 ARG INSTALL_CHANNEL=pytorch-nightly
 ENV CONDA_OVERRIDE_CUDA=${CUDA_VERSION}
 RUN /opt/conda/bin/conda install -c "${INSTALL_CHANNEL}" -c "${CUDA_CHANNEL}" -y python=${PYTHON_VERSION} pytorch torchvision torchtext "cudatoolkit=${CUDA_VERSION}" && \

docker.Makefile

diff --git a/docker.Makefile b/docker.Makefile
index e9a142c..5b45baf 100644
--- a/docker.Makefile
+++ b/docker.Makefile
@@ -8,7 +8,7 @@
 DOCKER_ORG                = $(shell whoami)
 endif
 
-CUDA_VERSION              = 11.0
+CUDA_VERSION              = 11.1
 CUDNN_VERSION             = 8
 BASE_RUNTIME              = ubuntu:18.04
 BASE_DEVEL                = nvidia/cuda:$(CUDA_VERSION)-cudnn$(CUDNN_VERSION)-devel-ubuntu18.04