FROM nvcr.io/nvidia/pytorch:22.05-py3

[glenn-jocher]

🛠️ PR Summary

_{Made with ❤️ by Ultralytics Actions}

🌟 Summary

Update to the Dockerfile for YOLOv5 to use a newer PyTorch base image.

📊 Key Changes

• The base image in the Dockerfile was updated from nvcr.io/nvidia/pytorch:22.04-py3 to nvcr.io/nvidia/pytorch:22.05-py3.
• A large directory (/opt/pytorch) is removed within the image to reduce its size.

🎯 Purpose & Impact

• Purpose: To keep the YOLOv5 Docker environment up to date with the latest NVIDIA PyTorch container, ensuring compatibility with the most recent features and optimizations.
• Impact: Users will benefit from potential performance improvements and bug fixes in the newer PyTorch version. Additionally, the removal of a large directory will result in a lighter Docker image, which can decrease download times and save disk space.