Compatibility Issue

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/ 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

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