Hugging Face's logo

Spaces:
xiang-wuu
/
yolov5
Runtime error

App Files Community
glenn-jocher commited on
Commit
7c2691f
·
unverified ·
1 Parent(s): 6e46617

FROM nvcr.io/nvidia/pytorch:22.05-py3 (#8162)

Browse files
Files changed (1) hide show
  1. utils/docker/Dockerfile +1 -1
utils/docker/Dockerfile CHANGED
@@ -3,7 +3,7 @@
3
  # Image is CUDA-optimized for YOLOv5 single/multi-GPU training and inference
4
 
5
  # Start FROM NVIDIA PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
6
- FROM nvcr.io/nvidia/pytorch:22.04-py3
7
  RUN rm -rf /opt/pytorch # remove 1.2GB dir
8
 
9
  # Downloads to user config dir
 
3
  # Image is CUDA-optimized for YOLOv5 single/multi-GPU training and inference
4
 
5
  # Start FROM NVIDIA PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
6
+ FROM nvcr.io/nvidia/pytorch:22.05-py3
7
  RUN rm -rf /opt/pytorch # remove 1.2GB dir
8
 
9
  # Downloads to user config dir