-
Notifications
You must be signed in to change notification settings - Fork 26.3k
Description
🚀 [RFC] Cuda support matrix for Release 2.4
Opening this RFC to discuss CUDA version support for future PyTorch releases:
Option 1 - CUDA 11 and CUDA 12:
CUDA 11.8, CUDNN 8.9.7.29
CUDA 12.4, CUDNN 8.9.7.29 - Version hosted on pypi
Option 2 - CUDA 12:
CUDA 12.1, CUDNN 8.9.7.29
CUDA 12.4, CUDNN 8.9.7.29 - Version hosted on pypi
Option 3
CUDA 11.8, CUDNN 8.9.7.29
CUDA 12.1, CUDNN 8.9.7.29 - Version hosted on pypi, as stable
CUDA 12.4, CUDNN 8.9.7.29 - Experimental version
(Please note CUDNN version listed here 8.9.7.29 is not final, we may upgrade it for 2.4 release)
One advantage of Option 1 is the fact that older cuda driver is not compatible with CUDA 12 hence people with older drivers will benefit latest pytorch.
Please refer to:
https://docs.nvidia.com/deploy/cuda-compatibility/index.html#minor-version-compatibility
| CUDA Toolkit | Linux x86_64 Minimum Required Driver Version | Windows Minimum Required Driver Version |
|---|---|---|
| CUDA 12.x | >=525.60.13 | >=527.41 |
| CUDA 11.x | >= 450.80.02* | >=452.39* |