系统:ubuntu 18.04
显卡:3070
cuda11.1
torch1.8.0
试过很多版本,但是都会报错
RuntimeError:Error compiling object for extension
经过多方尝试最终找到一个合适的DCNv2源码:链接: 点击这里.
然后进入安装的虚拟环境进行下面的操作
当然这一步如果已经安装好这些包就不要运行了
pip install torch torchvision torchaudio
这个是要运行一下的
sudo apt-get install ninja-build
最后执行
python3 setup.py build develop
参考:
[1] https://github.com/tteepe/DCNv2
[2] https://github.com/jinfagang/DCNv2_latest



