|
|
|
|
|
|
|
|
# Image is CUDA-optimized for YOLOv5 single/multi-GPU training and inference |
|
|
# 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 |
|
|
# Start FROM NVIDIA PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch |
|
|
FROM nvcr.io/nvidia/pytorch:22.04-py3 |
|
|
|
|
|
|
|
|
FROM nvcr.io/nvidia/pytorch:22.05-py3 |
|
|
RUN rm -rf /opt/pytorch # remove 1.2GB dir |
|
|
RUN rm -rf /opt/pytorch # remove 1.2GB dir |
|
|
|
|
|
|
|
|
# Downloads to user config dir |
|
|
# Downloads to user config dir |