PyTorch 2.2 with CUDA 12.1
For PyTorch 2.2 with CUDA 12.1, pytorch release is 2.2; cuda build is CUDA 12.1; python versions is >=3.8, <=3.11, (3.12 experimental); cudnn version is 8.9.2.26; 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.2 with CUDA 12.1 our reading
- PyTorch release
- 2.2 verified
- CUDA build
- CUDA 12.1 verified
- Python versions
- >=3.8, <=3.11, (3.12 experimental) verified
- cuDNN version
- 8.9.2.26 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.2 | >=3.8, <=3.11, (3.12 experimental) | C++17 | CUDA 11.8, CUDNN 8.7.0.84 | CUDA 12.1, CUDNN 8.9.2.26 | ROCm 5.7 |
— 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.2/manywheel/build_cuda.sh
- raw.githubusercontent.comhttps://raw.githubusercontent.com/pytorch/builder/release/2.2/windows/cuda121.bat
- docs.nvidia.comhttps://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html