FROM nvcr.io/nvidia/pytorch:22.07-py3 (#8808)
This commit is contained in:
1 parent
9559601b9a
commit
555976b346
1 file changed
+1
-1
@@ -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.06-py3
|
||||
FROM nvcr.io/nvidia/pytorch:22.07-py3
|
||||
RUN rm -rf /opt/pytorch # remove 1.2GB dir
|
||||
|
||||
# Downloads to user config dir
|
||||
|
||||
Reference in new issue
Block a user