Skip to content

feat(torch): Update to PyTorch 2.7 & CUDA 12.8.1 #93

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

Merged
merged 10 commits into from
May 13, 2025
Merged

Conversation

Eta0
Copy link
Collaborator

@Eta0 Eta0 commented May 13, 2025

PyTorch, CUDA, and NCCL updates

This change updates the following components:

  • PyTorch 2.6.0 → 2.7.0
  • CUDA 12.8.0 → 12.8.1
  • NCCL 2.25.1-1 → 2.26.5-1
    Along with a few other miscellaneous component updates.

This also fixes a few build issues (e.g. installing missing runtime libs, pinning the version of CMake used) and improves robustness (e.g. using multiple methods to ensure that sm_100 is always a targeted arch for builds).

@Eta0 Eta0 requested a review from sangstar May 13, 2025 20:17
@Eta0 Eta0 self-assigned this May 13, 2025
@Eta0 Eta0 added bug Something isn't working enhancement New feature or request labels May 13, 2025
Copy link
Contributor

@sangstar sangstar left a comment

Choose a reason for hiding this comment

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

LGTM; couple questions.

@Eta0 Eta0 merged commit bfbb68e into main May 13, 2025
54 of 56 checks passed
@Eta0 Eta0 deleted the es/torch-v2.7.0 branch May 13, 2025 20:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants