FROM nvcr.io/nvidia/pytorch:21.10-py3 ()

Reordered installation may help reduce resource usage in autobuild
pull/6677/head
Glenn Jocher 2022-02-17 15:42:47 +01:00 committed by GitHub
parent 25456284b4
commit 18b5081166
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -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