GPU compute capability support for each pytorch version

5 
$ python -c "import torch; print(torch.cuda.get_arch_list())"

Returns this error:

Traceback (most recent call last):
  File "<string>", line 1, in <module>
AttributeError: module 'torch.cuda' has no attribute 'get_arch_list'

Env:

Python 3.7.9 (default, Aug 31 2020, 12:42:55) 
[GCC 7.3.0] :: Anaconda, Inc. on linux

torch: 1.4.0     torchvision: 0.5.0

Cuda compilation tools, release 11.2, V11.2.142
Build cuda_11.2.r11.2/compiler.29558016_0
1 Like