30系列显卡编译可变性卷积(DCNv2)

配置

系统:ubuntu 18.04
显卡:3070
cuda11.1
torch1.8.0

编译DCNv2

试过很多版本,但是都会报错

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

你可能感兴趣的:(python,人工智能,神经网络)