Skip to content

Commit 1d61d79

Browse files
authored
Update CUDA_UPGRADE_GUIDE.MD
1 parent 3ae93de commit 1d61d79

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

CUDA_UPGRADE_GUIDE.MD

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# CUDA Upgrade Guide/Runbook
1+
# CUDA/CUDNN Upgrade Runbook
22

33
So you wanna upgrade PyTorch to support a new CUDA? Follow these steps in order! They are adapted from previous CUDA upgrade processes.
44

@@ -117,4 +117,11 @@ propagate the CI changes so that torchvision and torchaudio can be packaged for
117117
4. Almost every change in the above sample is copy-pasted from either itself or other existing parts of code in the
118118
builder repo. The difficulty again is not changing the config but rather verifying and debugging any failing builds.
119119

120-
Congrats! PyTorch now has support for a new CUDA version and you made it happen!
120+
This completes CUDA and CUDNN upgrade. Congrats! PyTorch now has support for a new CUDA version and you made it happen!
121+
122+
## Upgrade CUDNN version only
123+
124+
If you require to update CUDNN version for already existing CUDA version, please perform the followin modifications.
125+
1. Builder PR: https://github.com/pytorch/builder/pull/1271
126+
2. Add new cudnn vesion to windows AMI: https://github.com/pytorch/test-infra/pull/1523. Rebuild and retest the AMI. Follow step 6 Generate new Windows AMI, test and deploy to canary and prod.
127+
3. Create PyTorch PR: https://github.com/pytorch/pytorch/pull/93086

0 commit comments

Comments
 (0)