From b54fd0ac28b921756d8eaa66cdd12f1ba55833df Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 26 Mar 2023 20:27:43 +0200 Subject: [PATCH] FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime (#11246) Signed-off-by: Glenn Jocher --- utils/docker/Dockerfile | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/utils/docker/Dockerfile b/utils/docker/Dockerfile index b5d2af9..811ad4a 100644 --- a/utils/docker/Dockerfile +++ b/utils/docker/Dockerfile @@ -2,9 +2,8 @@ # Builds ultralytics/yolov5:latest image on DockerHub https://hub.docker.com/r/ultralytics/yolov5 # Image is CUDA-optimized for YOLOv5 single/multi-GPU training and inference -# Start FROM NVIDIA PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch -# FROM docker.io/pytorch/pytorch:latest -FROM pytorch/pytorch:latest +# Start FROM PyTorch image https://hub.docker.com/r/pytorch/pytorch +FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime # Downloads to user config dir ADD https://ultralytics.com/assets/Arial.ttf https://ultralytics.com/assets/Arial.Unicode.ttf /root/.config/Ultralytics/