ValueError: Unknown CUDA arch (8.6) or GPU not supported

tianzhi0549 / FCOS

FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)

https://arxiv.org/abs/1904.01355

Other

3.27k stars 630 forks source link

Open ghost opened 2 years ago

ghost commented 2 years ago

When run python setup.py build develop --no-deps I got the following error: ValueError: Unknown CUDA arch (8.6) or GPU not supported

How can i fix this?

hnanacc commented 1 year ago

Hi authors can we solve this? I'm also facing the same issue.

hnanacc commented 1 year ago

@YuxuanWen-Code this is more a pytorch and cuda issue. So try using newer versions of CUDA >= 11.1 and the problem will be solved.