Unsupported gpu architecture 'compute_86'

CHONSPQX / modulated-deform-conv

deformable convolution 2D 3D DeformableConvolution DeformConv Modulated Pytorch CUDA

MIT License

133 stars 17 forks source link

Unsupported gpu architecture 'compute_86' #12

Open DIVE128 opened 2 years ago

DIVE128 commented 2 years ago

Thanks for your great job. Our environment is linux with gpu RTX3090 and the fixed lowest version of pytorch is 1.7 with cuda11.0. When we run python setup.py install , it prompts nvcc fatal : Unsupported gpu architecture 'compute_86' . Is there any setting unsuitable?

ccl-private commented 2 years ago

export TORCH_CUDA_ARCH_LIST="7.5" in the terminal can fix this.