OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00
Code Issues Actions 51 Packages Projects Releases Wiki Activity

[CUDA] Upgrade cuDNN to 9.15.1 for CUDA 13 builds (#169412)

Browse Source 
Opening this PR for testing...

Note that we are proposing 9.15 instead of 9.16 as we have not had sufficient signal on 9.16 internally

NS: Added hacky workaround to install 9.15.1 for torchbench testing

Pull Request resolved: https://github.com/pytorch/pytorch/pull/169412
Approved by: https://github.com/atalman, https://github.com/malfet

Co-authored-by: Nikita Shulga <2453524+malfet@users.noreply.github.com>
Co-authored-by: Ting Lu <tingl@nvidia.com>
This commit is contained in:
Nikita Shulga
2025-12-23 04:55:48 +00:00
committed by PyTorch MergeBot
parent 60fa9d8ad0
commit bfa6f5e073
5 changed files with 21 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.github/scripts/generate_binary_build_matrix.py vendored
View File

@@ -108,7 +108,7 @@ PYTORCH_EXTRA_INSTALL_REQUIREMENTS = {
"nvidia-cuda-nvrtc==13.0.88; platform_system == 'Linux' | "
"nvidia-cuda-runtime~=13.0.48; platform_system == 'Linux' | "
"nvidia-cuda-cupti==13.0.85; platform_system == 'Linux' | "
"nvidia-cudnn-cu13==9.13.0.50; platform_system == 'Linux' | "
"nvidia-cudnn-cu13==9.15.1.9; platform_system == 'Linux' | "
"nvidia-cublas==13.1.0.3; platform_system == 'Linux' | "
"nvidia-cufft==12.0.0.61; platform_system == 'Linux' | "
"nvidia-curand==10.4.0.35; platform_system == 'Linux' | "