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

[glenn-jocher]

🛠️ PR Summary

_{Made with ❤️ by Ultralytics Actions}

🌟 Summary

Update to the Dockerfile to use a newer NVIDIA PyTorch base image.

📊 Key Changes

• Upgraded the base image from nvcr.io/nvidia/pytorch:21.05-py3 to nvcr.io/nvidia/pytorch:21.07-py3.

🎯 Purpose & Impact

• Purpose: To ensure the Docker environment for YOLOv5 leverages up-to-date features and improvements from a newer PyTorch version provided by NVIDIA.
• Impact: Users may experience improved performance, better compatibility, and access to newer software updates with the updated PyTorch base image. It also ensures better support for CUDA and GPU acceleration for machine learning tasks. 🚀