PyTorch 2.1 with CUDA 11.8
For PyTorch 2.1 with CUDA 11.8, pytorch release is 2.1; cuda build is CUDA 11.8; python versions is >=3.8, <=3.11; cudnn version is 8.7.0.84; 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.1 with CUDA 11.8 our reading
- PyTorch release
- 2.1 verified
- CUDA build
- CUDA 11.8 verified
- Python versions
- >=3.8, <=3.11 verified
- cuDNN version
- 8.7.0.84 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};3.7;9.0" verified
- GPU architectures compiled in, Windows wheel
- 3.7+PTX;5.0;6.0;6.1;7.0;7.5;8.0;8.6;9.0 verified
- Minimum driver (minor version compatibility)
- >= 450 verified
- Toolkit driver, Linux x86_64
- >=520.61.05 verified
- Toolkit driver, Windows x86_64
- >=520.06 verified
- Wheel index URL
- https://download.pytorch.org/whl/cu118 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.1 | >=3.8, <=3.11 | C++17 | CUDA 11.8, CUDNN 8.7.0.84 | CUDA 12.1, CUDNN 8.9.2.26 | ROCm 5.6 |
— 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
11.8) TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST};3.7;9.0"
GPU architectures compiled in, Windows wheel
IF "%BUILD_VISION%" == "" ( set TORCH_CUDA_ARCH_LIST=3.7+PTX;5.0;6.0;6.1;7.0;7.5;8.0;8.6;9.0
Minimum driver (minor version compatibility)
11.x | >= 450 | < 525
Toolkit driver, Linux x86_64 and Toolkit driver, Windows x86_64
CUDA 11.8 GA | >=520.61.05 | >=520.06
— 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.1/manywheel/build_cuda.sh
- raw.githubusercontent.comhttps://raw.githubusercontent.com/pytorch/builder/release/2.1/windows/cuda118.bat
- docs.nvidia.comhttps://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html