Compatibility Issue

1

Hi, I build pytorch from source in a docker container where the CUDA is 11.1. But I can not use it in A100, I got the following error:

/opt/conda/lib/python3.8/site-packages/torch/cuda/__init__.py:145: UserWarning: 
A100-SXM4-40GB with CUDA capability sm_80 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_75.
If you want to use the A100-SXM4-40GB GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

image
image1762×2168 649 KB

2

OK, I solved this by passing exact CUDA_ARCH when building pytorch from source TORCH_CUDA_ARCH_LIST="8.0" python setup.py install