PyTorch 2.4 with CUDA 12.4
For PyTorch 2.4 with CUDA 12.4, pytorch release is 2.4; cuda build is CUDA 12.4; python versions is >=3.8, <=3.12; 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.4 with CUDA 12.4 our reading
- PyTorch release
- 2.4 verified
- CUDA build
- CUDA 12.4 verified
- Python versions
- >=3.8, <=3.12 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
- >=550.54.14 verified
- Toolkit driver, Windows x86_64
- >=551.61 verified
- Wheel index URL
- https://download.pytorch.org/whl/cu124 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.4 | >=3.8, <=3.12 | C++17 | CUDA 11.8, CUDA 12.1, CUDNN 9.1.0.70 | CUDA 12.4, CUDNN 9.1.0.70 | ROCm 6.1 |
— 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.4) if [[ "$GPU_ARCH_TYPE" = "cuda-aarch64" ]]; then TORCH_CUDA_ARCH_LIST="9.0" else TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST};9.0" fi
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.4 GA | >=550.54.14 | >=551.61
— 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.4/manywheel/build_cuda.sh
- raw.githubusercontent.comhttps://raw.githubusercontent.com/pytorch/builder/release/2.4/windows/cuda124.bat
- docs.nvidia.comhttps://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html