FROM nvcr.io/nvidia/pytorch:21.04-py3 (ultralytics#3030)

(cherry picked from commit 4c7d686)
Lechtr · May 24, 2021 · 5d67514 · 5d67514
1 parent 9f80fde
commit 5d67514
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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