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

[glenn-jocher]

🛠️ PR Summary

_{Made with ❤️ by Ultralytics Actions}

🌟 Summary

Update to Dockerfile for Enhanced Compatibility

📊 Key Changes

• 📦 Updated the base image from Nvidia PyTorch 21.05-py3 to 21.10-py3.
• 🛠 Installed newer versions of torch and torchvision along with numpy and Pillow.
• 🧹 Cleaned up unnecessary installations of nvidia-tensorboard and nvidia-tensorboard-plugin-dlprof.

🎯 Purpose & Impact

• 🚀 Purpose: To provide a more current software environment for users, enhancing compatibility with the latest features and security patches.
• ✨ Impact: Users can expect better performance and improved integration with the latest machine learning libraries and tools. This may lead to easier deployments and a smoother development experience for YOLOv5-related projects.