From 5d67514780cb28a35b4d370e0db24171c70a625c Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 5 May 2021 00:16:35 +0200
Subject: [PATCH] FROM nvcr.io/nvidia/pytorch:21.04-py3 (#3030)

(cherry picked from commit 4c7d686d7e524736c69b193441a64376140ba085)
---
 Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index b47e5bbff194..423c9c5b37f9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,5 @@
 # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
-FROM nvcr.io/nvidia/pytorch:21.03-py3
+FROM nvcr.io/nvidia/pytorch:21.04-py3
 
 # Install linux packages
 RUN apt update && apt install -y zip htop screen libgl1-mesa-glx