Skip to content

An unofficial list of supported compute capability by each release of PyTorch (linux)

Notifications You must be signed in to change notification settings

evelthon/PyTorch-supported-compute-capability

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 

Repository files navigation

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

About

An unofficial list of supported compute capability by each release of PyTorch (linux)

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages