Hugging Face's logo

Spaces:
xiang-wuu
/
yolov5
Runtime error

App Files Community
glenn-jocher commited on
Commit
18b5081
·
unverified ·
1 Parent(s): 2545628

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

Browse files

Reordered installation may help reduce resource usage in autobuild

Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -1,7 +1,7 @@
1
  # YOLOv5 🚀 by Ultralytics, GPL-3.0 license
2
 
3
  # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
4
- FROM nvcr.io/nvidia/pytorch:21.10-py3
5
 
6
  # Install linux packages
7
  RUN apt update && apt install -y zip htop screen libgl1-mesa-glx
 
1
  # YOLOv5 🚀 by Ultralytics, GPL-3.0 license
2
 
3
  # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
4
+ FROM nvcr.io/nvidia/pytorch:22.01-py3
5
 
6
  # Install linux packages
7
  RUN apt update && apt install -y zip htop screen libgl1-mesa-glx