search

lucasjinreal / DCNv2_latest

DCNv2 supports decent pytorch such as torch 1.5+ (now 1.8+)
BSD 3-Clause "New" or "Revised" License
626 stars 124 forks source link

nvcc fatal : Unsupported gpu architecture 'compute_86' #40

Open benjamin666 opened 3 years ago

benjamin666 commented 3 years ago

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

AiueoABC commented 3 years ago

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 commented 5 months ago

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