PyTorch-supported Nvidia compute capability

License: CC BY-NC-SA 4.0 (https://creativecommons.org/licenses/by-nc-sa/4.0/)

The outcome

(Table is not sorted)

wheel file	Compute capability
torch-1.3.0+cu100-cp35-cp35m-linux_x86_64.whl	30, 35, 50, 60, 61, 70, 75
torch-1.3.0+cu100-cp36-cp36m-linux_x86_64.whl	30, 35, 50, 60, 61, 70, 75
torch-1.3.0+cu100-cp37-cp37m-linux_x86_64.whl	35, 50, 60, 61, 70, 75
torch-1.3.1+cu100-cp35-cp35m-linux_x86_64.whl	37, 50, 60, 61, 70, 75
torch-1.3.1+cu100-cp36-cp36m-linux_x86_64.whl	30, 35, 37, 50, 60, 61, 70, 75
torch-1.3.1+cu100-cp37-cp37m-linux_x86_64.whl	30, 35, 37, 50, 60, 61, 70, 75
torch-1.4.0+cu100-cp35-cp35m-linux_x86_64.whl	30, 35, 37, 50, 60, 61, 70, 75
torch-1.4.0+cu100-cp36-cp36m-linux_x86_64.whl	30, 35, 37, 50, 60, 61, 70, 75
torch-1.4.0+cu100-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 61, 70, 75
torch-1.4.0+cu100-cp38-cp38-linux_x86_64.whl	37, 50, 60, 61, 70, 75
torch-1.5.0+cu101-cp35-cp35m-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75
torch-1.5.0+cu101-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 61, 70, 75
torch-1.5.0+cu101-cp37-cp37m-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75
torch-1.5.0+cu101-cp38-cp38-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75
torch-1.5.1+cu101-cp35-cp35m-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75
torch-1.5.1+cu101-cp36-cp36m-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75
torch-1.5.1+cu101-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 61, 70, 75
torch-1.5.1+cu101-cp38-cp38-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75
torch-1.6.0+cu101-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70, 75
torch-1.6.0+cu101-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70, 75
torch-1.6.0+cu101-cp38-cp38-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75
torch-1.7.0+cu101-cp36-cp36m-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75
torch-1.7.0+cu101-cp37-cp37m-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75
torch-1.7.0+cu101-cp38-cp38-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75
torch-1.7.1+cu101-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70, 75
torch-1.7.1+cu101-cp37-cp37m-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75
torch-1.7.1+cu101-cp38-cp38-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75
torch-1.7.1+cu101-cp39-cp39-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75
torch-1.8.0+cu101-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70
torch-1.8.0+cu101-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70
torch-1.8.0+cu101-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70
torch-1.8.0+cu101-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70
torch-1.8.1+cu101-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70
torch-1.8.1+cu101-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70
torch-1.8.1+cu101-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70
torch-1.8.1+cu101-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70
torch-1.10.0+cu102-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70
torch-1.10.0+cu102-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70
torch-1.10.0+cu102-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70
torch-1.10.0+cu102-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70
torch-1.10.1+cu102-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70
torch-1.10.1+cu102-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70
torch-1.10.1+cu102-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70
torch-1.10.1+cu102-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70
torch-1.8.1+cu102-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70
torch-1.8.1+cu102-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70
torch-1.8.1+cu102-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70
torch-1.8.1+cu102-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70
torch-1.9.0+cu102-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70
torch-1.9.0+cu102-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70
torch-1.9.0+cu102-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70
torch-1.9.0+cu102-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70
torch-1.9.1+cu102-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70
torch-1.9.1+cu102-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70
torch-1.9.1+cu102-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70
torch-1.9.1+cu102-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70
torch-1.7.0+cu110-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70, 75, 80
torch-1.7.0+cu110-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70, 75, 80
torch-1.7.0+cu110-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70, 75, 80
torch-1.7.1+cu110-cp36-cp36m-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75, 80
torch-1.7.1+cu110-cp37-cp37m-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75, 80
torch-1.7.1+cu110-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70, 75, 80
torch-1.7.1+cu110-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70, 75, 80
torch-1.10.0+cu111-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.0+cu111-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.0+cu111-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.0+cu111-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.1+cu111-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.1+cu111-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.1+cu111-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.1+cu111-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.8.0+cu111-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.8.0+cu111-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.8.0+cu111-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.8.0+cu111-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.8.1+cu111-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.8.1+cu111-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.8.1+cu111-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.8.1+cu111-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.9.0+cu111-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.9.0+cu111-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.9.0+cu111-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.9.0+cu111-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.9.1+cu111-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.9.1+cu111-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.9.1+cu111-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.9.1+cu111-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.0+cu113-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.0+cu113-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.0+cu113-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.0+cu113-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.1+cu113-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.1+cu113-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.1+cu113-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.1+cu113-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.2.0+cu92-cp35-cp35m-manylinux1_x86_64.whl	35, 50, 60, 61, 70
torch-1.2.0+cu92-cp36-cp36m-manylinux1_x86_64.whl	35, 50, 60, 61, 70
torch-1.2.0+cu92-cp37-cp37m-manylinux1_x86_64.whl	30, 35, 50, 60, 61, 70
torch-1.3.0+cu92-cp35-cp35m-linux_x86_64.whl	35, 50, 60, 61, 70
torch-1.3.0+cu92-cp36-cp36m-linux_x86_64.whl	35, 50, 60, 61, 70
torch-1.3.0+cu92-cp37-cp37m-linux_x86_64.whl	30, 35, 50, 60, 61, 70
torch-1.3.1+cu92-cp35-cp35m-linux_x86_64.whl	37, 50, 60, 61, 70
torch-1.3.1+cu92-cp36-cp36m-linux_x86_64.whl	35, 37, 50, 60, 61, 70
torch-1.3.1+cu92-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 61, 70
torch-1.4.0+cu92-cp35-cp35m-linux_x86_64.whl	35, 37, 50, 60, 61, 70
torch-1.4.0+cu92-cp36-cp36m-linux_x86_64.whl	35, 37, 50, 60, 61, 70
torch-1.4.0+cu92-cp37-cp37m-linux_x86_64.whl	35, 37, 50, 60, 61, 70
torch-1.4.0+cu92-cp38-cp38-linux_x86_64.whl	37, 50, 60, 61, 70
torch-1.5.0+cu92-cp35-cp35m-linux_x86_64.whl	35, 37, 50, 60, 61, 70
torch-1.5.0+cu92-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 61, 70
torch-1.5.0+cu92-cp37-cp37m-linux_x86_64.whl	35, 37, 50, 60, 61, 70
torch-1.5.0+cu92-cp38-cp38-linux_x86_64.whl	37, 50, 60, 61, 70
torch-1.5.1+cu92-cp35-cp35m-linux_x86_64.whl	35, 37, 50, 60, 61, 70
torch-1.5.1+cu92-cp36-cp36m-linux_x86_64.whl	35, 37, 50, 60, 61, 70
torch-1.5.1+cu92-cp37-cp37m-linux_x86_64.whl	35, 37, 50, 60, 61, 70
torch-1.5.1+cu92-cp38-cp38-linux_x86_64.whl	35, 37, 50, 60, 61, 70
torch-1.6.0+cu92-cp36-cp36m-linux_x86_64.whl	35, 37, 50, 60, 61, 70
torch-1.6.0+cu92-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70
torch-1.6.0+cu92-cp38-cp38-linux_x86_64.whl	35, 37, 50, 60, 61, 70
torch-1.7.0+cu92-cp36-cp36m-linux_x86_64.whl	30, 35, 37, 50, 52, 60, 61, 70
torch-1.7.0+cu92-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70
torch-1.7.0+cu92-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70
torch-1.7.1+cu92-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70
torch-1.7.1+cu92-cp37-cp37m-linux_x86_64.whl	30, 35, 37, 50, 52, 60, 61, 70
torch-1.7.1+cu92-cp38-cp38-linux_x86_64.whl	30, 35, 37, 50, 52, 60, 61, 70
torch-1.7.1+cu92-cp39-cp39-linux_x86_64.whl	30, 35, 37, 50, 52, 60, 61, 70
torch-1.10.2+cu102-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70
torch-1.10.2+cu102-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70
torch-1.10.2+cu102-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70
torch-1.10.2+cu102-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70
torch-1.10.2+cu111-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.2+cu111-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.2+cu111-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.2+cu111-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.2+cu113-cp36-cp36m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.2+cu113-cp37-cp37m-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.2+cu113-cp38-cp38-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.10.2+cu113-cp39-cp39-linux_x86_64.whl	37, 50, 60, 70, 75, 80, 86
torch-1.11.0+cu102-cp310-cp310-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75
torch-1.11.0+cu102-cp37-cp37m-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75
torch-1.11.0+cu102-cp38-cp38-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75
torch-1.11.0+cu102-cp39-cp39-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75
torch-1.11.0+cu113-cp310-cp310-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75, 80, 86
torch-1.11.0+cu113-cp37-cp37m-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75, 80, 86
torch-1.11.0+cu113-cp38-cp38-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75, 80, 86
torch-1.11.0+cu113-cp39-cp39-linux_x86_64.whl	35, 37, 50, 60, 61, 70, 75, 80, 86
torch-1.11.0+cu115-cp310-cp310-linux_x86_64.whl	35, 37, 50, 60, 70, 75, 80, 86
torch-1.11.0+cu115-cp37-cp37m-linux_x86_64.whl	35, 37, 50, 60, 70, 75, 80, 86
torch-1.11.0+cu115-cp38-cp38-linux_x86_64.whl	35, 37, 50, 60, 70, 75, 80, 86
torch-1.11.0+cu115-cp39-cp39-linux_x86_64.whl	35, 37, 50, 60, 70, 75, 80, 86

The why

It basically all started when I was trying to use an old nvidia graphics card with PyTorch. I would get a message that it was not supported with the latest version. As such, two solutions were possible

1. Compile from source
2. Install an older version that supports the desired compute capability

This attempts to answer the second question.

The steps followed are:

1. Usewget to download (linux) PyTorch wheel file w/ CUDA support.
2. Execute a python script (generate.py) to generate the list (to use it, update the variable WHL_FILE_FOLDER to point to the folder where the wheel files are stored).

Note: cuobjdump command must be available

How to download wheel files

wget -r --no-parent -A "torch-*cu*cp3*linux*.whl" -nd -c https://download.pytorch.org/whl/torch_stable.html

Attributions

• https://discuss.pytorch.org/t/gpu-compute-capability-support-for-each-pytorch-version/62434