PyTorch 2.5 with CUDA 12.1
For PyTorch 2.5 with CUDA 12.1, pytorch release is 2.5; cuda build is CUDA 12.1; python versions is >=3.9, <=3.12, (3.13 experimental); cudnn version is 9.1.0.70; build script: base architecture list is TORCH_CUDA_ARCH_LIST="5.0;6.0;7.0;7.5;8.0;8.6", recorded from its source on 2026-09-16.
- Pairing
- PyTorch 2.5 with CUDA 12.1 our reading
- PyTorch release
- 2.5 verified
- CUDA build
- CUDA 12.1 verified
- Python versions
- >=3.9, <=3.12, (3.13 experimental) verified
- cuDNN version
- 9.1.0.70 verified
- Build script: base architecture list
- TORCH_CUDA_ARCH_LIST="5.0;6.0;7.0;7.5;8.0;8.6" verified
- Build script: this CUDA build's architecture line
- TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST};9.0" verified
- GPU architectures compiled in, Windows wheel
- 5.0;6.0;6.1;7.0;7.5;8.0;8.6;9.0 verified
- Minimum driver (minor version compatibility)
- >= 525 verified
- Toolkit driver, Linux x86_64
- >=530.30.02 verified
- Toolkit driver, Windows x86_64
- >=531.14 verified
- Wheel index URL
- https://download.pytorch.org/whl/cu121 our reading
Values marked our reading are our classification of what the source says — the source does not print them in those words. The quote below is the evidence for each one; judge it yourself.
What the source says
| 2.5 | >=3.9, <=3.12, (3.13 experimental) | C++17 | CUDA 11.8, CUDA 12.1, CUDA 12.4, CUDNN 9.1.0.70 | None | ROCm 6.2 |
— raw.githubusercontent.com, retrieved 2026-09-16
Where each value comes from
This source states these in separate places, so each value is shown with the passage that states it.
Build script: base architecture list
TORCH_CUDA_ARCH_LIST="5.0;6.0;7.0;7.5;8.0;8.6"
Build script: this CUDA build's architecture line
12.1) TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST};9.0"
GPU architectures compiled in, Windows wheel
IF "%BUILD_VISION%" == "" ( set TORCH_CUDA_ARCH_LIST=5.0;6.0;6.1;7.0;7.5;8.0;8.6;9.0
Minimum driver (minor version compatibility)
12.x | >= 525 | < 580
Toolkit driver, Linux x86_64 and Toolkit driver, Windows x86_64
CUDA 12.1 GA | >=530.30.02 | >=531.14
— all from raw.githubusercontent.com, retrieved 2026-09-16
Sources
- raw.githubusercontent.comhttps://raw.githubusercontent.com/pytorch/pytorch/main/RELEASE.md
- raw.githubusercontent.comhttps://raw.githubusercontent.com/pytorch/builder/release/2.5/manywheel/build_cuda.sh
- raw.githubusercontent.comhttps://raw.githubusercontent.com/pytorch/builder/release/2.5/windows/cuda121.bat
- docs.nvidia.comhttps://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html