-
Notifications
You must be signed in to change notification settings - Fork 3.6k
docker: build images for latest PT 2.8
#21042
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
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Pull Request Overview
This PR updates Docker configurations to support PyTorch version 2.8, preparing the infrastructure for GPU CI testing with the latest PyTorch release. The changes involve updating version parameters across multiple Dockerfiles and GitHub Actions workflows.
- Updates PyTorch version from 2.0/2.1/2.7 to 2.8 across Docker configurations
- Updates CUDA version from 11.8.0 to 12.6.3 for compatibility with PyTorch 2.8
- Modifies build matrix in GitHub Actions to include PyTorch 2.8.0 configurations
Reviewed Changes
Copilot reviewed 4 out of 4 changed files in this pull request and generated no comments.
| File | Description |
|---|---|
| dockers/release/Dockerfile | Updates PyTorch version to 2.8 and CUDA version to 12.6.3 |
| dockers/nvidia/Dockerfile | Updates NVIDIA PyTorch container version from 22.09 to 24.05 |
| dockers/base-cuda/Dockerfile | Updates PyTorch version to 2.8 and improves MAKEFLAGS configuration |
| .github/workflows/docker-build.yml | Adds PyTorch 2.8.0 to build matrix and updates version configurations |
⚡ Required checks status: All passing 🟢Groups summary🟢 pytorch_lightning: Docker
These checks are required after the changes to Thank you for your contribution! 💜
|
What does this PR do?
preparing images to be able to bump GPU CI for testing with latest Torch
Before submitting
PR review
Anyone in the community is welcome to review the PR.
Before you start reviewing, make sure you have read the review guidelines. In short, see the following bullet-list:
Reviewer checklist
📚 Documentation preview 📚: https://pytorch-lightning--21042.org.readthedocs.build/en/21042/