Add CUDA 12.8 installation and Linux CD Docker images#145557
Add CUDA 12.8 installation and Linux CD Docker images#145557tinglvv wants to merge 2 commits intopytorch:mainfrom
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/145557
Note: Links to docs will display an error until the docs builds have been completed. ❌ 5 New Failures, 10 Cancelled JobsAs of commit 04a77bd with merge base bdc2c2a ( NEW FAILURES - The following jobs have failed:
CANCELLED JOBS - The following jobs were cancelled. Please retry:
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
|
Error Seems it is not possible to do all docker changes at once. Let me break this into installation script, then magma/libtorch update. |
| ldconfig | ||
| } | ||
|
|
||
| function prune_128 { |
There was a problem hiding this comment.
We don't ship cuDNN or cuBLAS in our binaries, so do we need to nvprune anything?
|
Closing to follow up in #145567. |
Breaking #145557 into two parts. Need to have manylinux-cuda12.8 in order to build magma. Issue: #145570 Pull Request resolved: #145567 Approved by: https://github.com/nWEIdia, https://github.com/atalman
Breaking pytorch#145557 into two parts. Need to have manylinux-cuda12.8 in order to build magma. Issue: pytorch#145570 Pull Request resolved: pytorch#145567 Approved by: https://github.com/nWEIdia, https://github.com/atalman

Add CUDA 12.8 installation script
Add Magma for CUDA 12.8
Add to Linux sbsa/x86 manywheel and libtorch Dockers
TODO: Cudnn Update to 9.7.0 once it is available