From 66cc50533985e55486836cd2a84b1d38269cf843 Mon Sep 17 00:00:00 2001
From: Jirka Borovec <Borda@users.noreply.github.com>
Date: Mon, 2 Aug 2021 18:05:36 +0200
Subject: [PATCH] update NGC (#8652)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* update NGC

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
---
 dockers/nvidia/Dockerfile | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/dockers/nvidia/Dockerfile b/dockers/nvidia/Dockerfile
index 88b558822f..9cab659eb0 100644
--- a/dockers/nvidia/Dockerfile
+++ b/dockers/nvidia/Dockerfile
@@ -12,8 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+ARG PYTORCH_VERSION=21.07
+
 # https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes
-FROM nvcr.io/nvidia/pytorch:21.06-py3
+FROM nvcr.io/nvidia/pytorch:${PYTORCH_VERSION}-py3
 
 LABEL maintainer="PyTorchLightning <https://github.com/PyTorchLightning>"