这个问题困扰了我半个多月了,在网上找了很多的资料,网上的解决办法也试了,但是都没用,今天终于解决了
PyTorch 版本: 1.4.0
CUDA 是否可用: True
CUDA 版本: 10.1
CuDNN 版本: 7501
DCNv2:https://github.com/CharlesShang/DCNv2
下载后进入 ../src/lib/models/networks/DCNv2目录下
运行
(CenterNet_Py37) D:\Thesis source code\CenterNet\src\lib\models\networks\DCNv2>python setup.py build install
我第一次编译之后还是找不到_ext模块的,也就是import _ext as _backbone还是报错
紧接删除旧的编译文件:
(CenterNet_Py37) D:\Thesis source code\CenterNet\src\lib\models\networks\DCNv2>python setup.py clean
running clean
removing 'build\temp.win-amd64-cpython-37' (and everything under it)
重新编译之后报错就解决了
(CenterNet_Py37) D:\Thesis source code\CenterNet\src\lib\models\networks\DCNv2>python setup.py build install