Unable to run on V100

#1

Hi, I build pytorch from source by TORCH_CUDA_ARCH_LIST="8.0" python setup.py install . But when running on GPU cluster, I encounter following error, Can anyone help me ? What should I do to make my pytorch compatible both in V100 and A100 ?

Found GPU5 Tesla V100-SXM2-32GB which is of cuda capability 7.0.
    PyTorch no longer supports this GPU because it is too old.
    The minimum cuda capability supported by this library is 7.5.

RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
#2

In your current build process you are only building PyTorch for compute capability 8.0 (Ampere).
If you want to add Volta, use TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0".

#3

Thanks so much! It solved all my problem