From dd3f02b9edbdf2ee5826c2b921cf6bcd71c90bbe Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 17 Feb 2022 15:42:47 +0100 Subject: [PATCH] FROM nvcr.io/nvidia/pytorch:21.10-py3 (#6673) Reordered installation may help reduce resource usage in autobuild --- Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index 33f428e81b1d..e691b4a9f08b 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,7 +1,7 @@ # YOLOv5 🚀 by Ultralytics, GPL-3.0 license # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch -FROM nvcr.io/nvidia/pytorch:21.10-py3 +FROM nvcr.io/nvidia/pytorch:22.01-py3 # Install linux packages RUN apt update && apt install -y zip htop screen libgl1-mesa-glx