Skip to content

Conversation

@glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Nov 9, 2025

Updated the base image to PyTorch 2.8.0 with CUDA 12.8.

🛠️ PR Summary

Made with ❤️ by Ultralytics Actions

🌟 Summary

Upgrade Docker base image to PyTorch 2.8.0 with CUDA 12.8 and cuDNN 9 for modern GPU compatibility and performance 🚀

📊 Key Changes

  • Update Docker base image from pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime to pytorch/pytorch:2.8.0-cuda12.8-cudnn9-runtime
  • Retain asset downloads (Arial fonts) to the Ultralytics config directory
  • Aligns container with current PyTorch/CUDA/cuDNN stack per latest PyTorch Docker tags

🎯 Purpose & Impact

  • Improved performance and compatibility on newer NVIDIA GPUs and drivers 💡
  • Access to PyTorch 2.8 features, optimizations, and security updates 🔒
  • Future-proofing: aligns with modern CUDA/cuDNN, reducing tech debt and easing maintenance 🧱
  • Potential breaking change: requires newer NVIDIA drivers (compatible with CUDA 12.x); users on older drivers may need to upgrade ⚠️
  • Possible image size/runtime behavior changes; validate workflows (training/inference) after updating ✅

Updated the base image to PyTorch 2.8.0 with CUDA 12.8.

Signed-off-by: Glenn Jocher <[email protected]>
@UltralyticsAssistant UltralyticsAssistant added dependencies Dependencies and packages devops GitHub Devops or MLops labels Nov 9, 2025
@UltralyticsAssistant
Copy link
Member

👋 Hello @glenn-jocher, thank you for submitting a ultralytics/yolov3 🚀 PR! To ensure a seamless integration of your work, please review the following checklist:

  • Define a Purpose: Clearly explain the purpose of your fix or feature in your PR description, and link to any relevant issues. Ensure your commit messages are clear, concise, and adhere to the project's conventions.
  • Synchronize with Source: Confirm your PR is synchronized with the ultralytics/yolov3 main branch. If it's behind, update it by clicking the 'Update branch' button or by running git pull and git merge main locally.
  • Ensure CI Checks Pass: Verify all Ultralytics Continuous Integration (CI) checks are passing. If any checks fail, please address the issues.
  • Update Documentation: Update the relevant documentation for any new or modified features.
  • Add Tests: If applicable, include or update tests to cover your changes, and confirm that all tests are passing.
  • Sign the CLA: Please ensure you have signed our Contributor License Agreement if this is your first Ultralytics PR by writing "I have read the CLA Document and I sign the CLA" in a new message.
  • Minimize Changes: Limit your changes to the minimum necessary for your bug fix or feature addition. "It is not daily increase but daily decrease, hack away the unessential. The closer to the source, the less wastage there is." — Bruce Lee

For more guidance, please refer to our Contributing Guide. This is an automated message; an Ultralytics engineer will assist you shortly. Thank you for contributing to Ultralytics! 🎉

Copy link
Member

@UltralyticsAssistant UltralyticsAssistant left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

🔍 PR Review

Made with ❤️ by Ultralytics Actions

Please reassess the base-image bump: PyTorch 2.8/CUDA 12.8 drops Maxwell (sm_5x) GPU support, so the new Dockerfile may break inference/training on older cards unless we have an alternative plan.

💬 Posted 1 inline comment

@glenn-jocher glenn-jocher merged commit 2c48b70 into master Nov 9, 2025
6 checks passed
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-1 branch November 9, 2025 18:32
@UltralyticsAssistant
Copy link
Member

🎉 Fantastic work, @glenn-jocher — PR merged! Upgrading the Docker base to PyTorch 2.8.0 with CUDA 12.8 and cuDNN 9 is a powerful step toward modern GPU performance and smoother maintenance.

“Inventing the future” starts here; as Alan Kay said, “The best way to predict the future is to invent it.” This change aligns us with the latest PyTorch Docker tags, unlocks new features and optimizations, and keeps YOLOv3 ready for today’s and tomorrow’s hardware.

Thank you for future-proofing the stack and keeping our users moving fast and forward. 🚀

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

dependencies Dependencies and packages devops GitHub Devops or MLops

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants