mirror of
https://github.com/ultralytics/yolov5.git
synced 2025-06-03 14:49:29 +08:00
FROM nvcr.io/nvidia/pytorch:22.09-py3 (#9711)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
parent
e4398cf179
commit
7f097ddb6c
@ -3,7 +3,7 @@
|
||||
# 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 nvcr.io/nvidia/pytorch:22.08-py3
|
||||
FROM nvcr.io/nvidia/pytorch:22.09-py3
|
||||
RUN rm -rf /opt/pytorch # remove 1.2GB dir
|
||||
|
||||
# Downloads to user config dir
|
||||
|
Loading…
x
Reference in New Issue
Block a user