当前位置:   article > 正文

编译DCNv3

编译dcnv3

编译DCNv3

INTERN-2.5

windows vs2017 cuda 11.6.2 pytorch 1.12.1+cu116

//修改 dcnv3_cuda.cu
//#include <torch/torch.h> //注释掉
torch:: 改成at::
//修改 dcnv3_cuda.h
#include <tuple>	//添加
  • 1
  • 2
  • 3
  • 4
  • 5

管理员权限打开Anaconda Prompt

cd InternImage-master\detection\ops_dcnv3
"C:\Program Files (x86)\Microsoft Visual Studio\2017\Professional\VC\Auxiliary\Build\vcvars64.bat"

set DISTUTILS_USE_SDK=1
python setup.py build install
  • 1
  • 2
  • 3
  • 4
  • 5
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小丑西瓜9/article/detail/93827
推荐阅读
相关标签
  

闽ICP备14008679号