DCNv2编译

基础环境

Win10操作系统
PyTorch1.5.1
CUDA10.2

下载DCNv2替换

DCNv2: 链接.

图片:

替换部分代码

\external\DCNv2\src\cuda\dcn_v2_cuda.cu
打开该dcn_v2_cuda.cu文件夹，将THCState_getCurrentStream(state)替换成c10::cuda::getCurrentCUDAStream()

打开cpp_extension.py文件

搜索match，进行如图所示的替换，在此仅替换了注视掉的一行

将cl.exe添加到系统环境变量

D:\worksoftware\Anaconda\envs\torch\lib\site-packages\torch\utils\cpp_extension.py:238: UserWarning: Error checking compiler version for cl: [WinError 2] 系统找不到指定的文件。
warnings.warn(‘Error checking compiler version for {}: {}’.format(compiler, error))
building ‘_ext’ extension
如果出现了这种报错，应该是vs没有在系统环境变量中，索引cl.exe所在的目录，将该目录添加到系统环境变量

添加后重启电脑，注销计算机后登陆才会生效

运行康康

cd到DCNv2目录，输入python setup.py build develop