使用DCNv2遇到的问题

git clone -b pytorch_1.7 https://github.com/ifzhang/DCNv2.git
cd DCNv2
./make.sh

搭建了虚拟环境，安装了pytorch=1.7.1，./make.sh遇到报错：

    raise ValueError("Unknown CUDA arch ({}) or GPU not supported".format(arch))
ValueError: Unknown CUDA arch (8.9) or GPU not supported

解决方法：在相应虚拟环境中的.../envs/fairmot_yopo/lib/python3.8/site-packages/torch/utils中的cpp_extension.py找到这一行

    supported_arches = ['3.5', '3.7', '5.0', '5.2', '5.3', '6.0', '6.1', '6.2',
                        '7.0', '7.2', '7.5', '8.0', '8.6']

在其中添加：'8.9'

修改后运行结果：

Installed ..../DCNv2
Processing dependencies for DCNv2==0.1
Finished processing dependencies for DCNv2==0.1