Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Bump torch>=1.8.0 and torchvision>=0.9.0 #11970

Merged
merged 1 commit into from
Aug 10, 2023
Merged

Bump torch>=1.8.0 and torchvision>=0.9.0 #11970

merged 1 commit into from
Aug 10, 2023

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Aug 10, 2023

πŸ€– Generated by Copilot at 243e103

Summary

πŸ“πŸ”₯πŸš€

This pull request updates the codebase and documentation to use PyTorch 1.8.0 as the minimum requirement for the yolov5 project. This ensures compatibility and performance with the latest PyTorch and torchvision packages.

Sing, O Muse, of the swift and skillful changes
That the wise and resourceful developers made
To the yolov5 project, the glorious offspring
Of the mighty and radiant PyTorch framework.

Walkthrough

  • Update PyTorch dependency to 1.8.0 and Python environment to 3.8.0 for the yolov5 project (link, link, link, link, link, link)
  • Change the CI testing matrix in .github/workflows/ci-testing.yml to use PyTorch 1.8.0 and Python 3.8.0 for the ubuntu-latest combination (link, link)
  • Update the README files in README.md and README.zh-CN.md to inform the users of the minimum PyTorch requirement and the recommended Python environment (link, link, link)
  • Remove the comment about torch 1.7.0 compatibility in models/experimental.py as it is no longer relevant (link)

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Update minimum PyTorch version from 1.7.0 to 1.8.0 across the repository.

πŸ“Š Key Changes

  • Updated Python CI testing to use PyTorch version 1.8.0 and torchvision 0.9.0.
  • Adjusted the Python and PyTorch version requirements in the README files accordingly.
  • Modified compatibility updates in models/experimental.py.

🎯 Purpose & Impact

  • Ensures compatibility with PyTorch version 1.8.0, providing users with the benefits of newer PyTorch features and optimizations. πŸ†•
  • Aligns the codebase with current best practices by using a more up-to-date version of PyTorch. βœ”οΈ
  • Enhances maintainability and support for users who have upgraded to newer versions of PyTorch. πŸ› οΈ

@glenn-jocher glenn-jocher merged commit 1a779ee into master Aug 10, 2023
10 checks passed
@glenn-jocher glenn-jocher deleted the torch_bump branch August 10, 2023 22:12
pleb631 pushed a commit to pleb631/yolov5 that referenced this pull request Jan 6, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

1 participant