nvcc fatal : Unsupported gpu architecture 'compute_86'

[benjamin666]

Someone meet this situation,How can I solve it? My GPU is 3090 and cuda driver is 11.4

[AiueoABC]

I found solution, this was the PyTorch issue for me. To fix this, first open /home/user/.local/lib/python3.6/site-packages/torch/utils/cpp_extension.py, and comment flags.append('-gencode=arch=compute_{},code=sm_{}'.format(num, num)) around ln1441 in cpp_extension.py.
Now, you can run make.sh with no trouble. Don't forget to modify "python" to "python3" in make.sh.

[hiepbk]

I found solution, this was the PyTorch issue for me. To fix this, first open /home/user/.local/lib/python3.6/site-packages/torch/utils/cpp_extension.py, and comment flags.append('-gencode=arch=compute_{},code=sm_{}'.format(num, num)) around ln1441 in cpp_extension.py. Now, you can run make.sh with no trouble. Don't forget to modify "python" to "python3" in make.sh.

It works for me. I use Docker Container with Pytorch 1.7.1, Ubuntu 20.04, Cuda v11.0