From 336ac9496faf51d8bdb0143fe53ee71ba1ebf1ca Mon Sep 17 00:00:00 2001 From: Elton Chen-Yu Ho Date: Tue, 1 Jun 2021 15:21:06 -0700 Subject: [PATCH] Fix mismatch in README.md Docker Image section (#59199) Summary: docker.Makefile has CUDNN_VERSION=8 as the defaults, but README.md states cuDNN v7 Pull Request resolved: https://github.com/pytorch/pytorch/pull/59199 Reviewed By: mruberry Differential Revision: D28808611 Pulled By: ngimel fbshipit-source-id: 96cea32bfe33184b2bff69b7bb7f3e50a2b9c6aa --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 99a0209ee9b..2d9c7ddacfc 100644 --- a/README.md +++ b/README.md @@ -353,7 +353,7 @@ should increase shared memory size either with `--ipc=host` or `--shm-size` comm **NOTE:** Must be built with a docker version > 18.06 -The `Dockerfile` is supplied to build images with Cuda support and cuDNN v7. +The `Dockerfile` is supplied to build images with CUDA 11.1 support and cuDNN v8. You can pass `PYTHON_VERSION=x.y` make variable to specify which Python version is to be used by Miniconda, or leave it unset to use the default. ```bash