PyTorch for Tesla k40c with cuda 11.2

4

You can specify multiple GPU architectures by building with:

TORCH_CUDA_ARCH_LIST="3.5 6.1" python setup.py install

where you can pass multiple architectures to the env var (as well as +PTX).