DCNv2编译

DCNv2编译

  • 基础环境
    • 下载DCNv2替换
    • 替换部分代码
    • 将cl.exe添加到系统环境变量
    • 运行康康

基础环境

Win10操作系统
PyTorch1.5.1
CUDA10.2

下载DCNv2替换

DCNv2: 链接.

图片:
DCNv2编译_第1张图片

替换部分代码

\external\DCNv2\src\cuda\dcn_v2_cuda.cu
打开该dcn_v2_cuda.cu文件夹,将THCState_getCurrentStream(state)替换成c10::cuda::getCurrentCUDAStream()
DCNv2编译_第2张图片
打开cpp_extension.py文件
在这里插入图片描述
搜索match,进行如图所示的替换,在此仅替换了注视掉的一行
DCNv2编译_第3张图片

将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所在的目录,将该目录添加到系统环境变量
DCNv2编译_第4张图片

DCNv2编译_第5张图片
添加后重启电脑,注销计算机后登陆才会生效

运行康康

cd到DCNv2目录,输入python setup.py build develop
DCNv2编译_第6张图片

你可能感兴趣的:(Yolact++,pytorch,深度学习)