win10安装detectron2

##1.安装支持cuda10.1的NVIDIA显卡驱动,一般最新的NVIDIA显卡驱动都支持
##2.如果目前cuda的版本不是10.1或10.1以上,就卸载掉,然后安装cuda10.1
##3.cuda10.1完成以后,除了软件自动添加上的系统环境变量以外,最好新增加以下三个:
添加系统环境变量:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib
##4.安装vs2019,社区版,专业版,企业版都行,推荐社区版,因为简单。
1)detectron2需要vs2019最新的编译器
2)如果之前电脑上已经装了vs的其他版本,重新设置vs2019的编译器,操作如下:
SET MSSdk=1
SET DISTUTILS_USE_SDK=1
call “C:/Program Files (x86)/Microsoft Visual Studio/2019/Enterprise/VC/Auxiliary/Build/vcvarsall.bat” amd64 -vcvars_ver=14.24
每个人的vcvarsall.bat路径不一样,找到自己的路径,然后替换掉上面的。
##5.安装最新版的anaconda3
##6.安装pytorch1.3.1,以下是pytorch官网给的指令:
pip install torch=1.3.1 torchvision=0.4.2 -f https://download.pytorch.org/whl/torch_stable.html
##7.安装cocoapi社区版:
下载以下链接的cocoapi,因为cocoapi官方暂不支持Windows
https://github.com/philferriere/cocoapi
然后用以下指令安装
cd coco/PythonAPI
python setup.py build_ext --inplace
python setup.py build_ext install
##8.下载fvcore:
下载链接:https://github.com/facebookresearch/fvcore
然后cd到setup.py所在目录,用以下指令安装
python setup.py build --force develop
##9.编译detectron2:
cd 到detectron2目录,找到setup.py文件,然后用以下指令编译:
python setup.py build --force develop

附言:
你可能还会遇到以下问题,我有贴对应的解决方案:
##1.编译时遇见:
1)
D:/detectron2/detectron2/layers/csrc/ROIAlign/ROIAlign_cuda.cu(337): error: no instance of function template “at::cuda::ATenCeilDiv” matches the argument list
argument types are: (long long, long)

D:/detectron2/detectron2/layers/csrc/ROIAlign/ROIAlign_cuda.cu(337): error: no instance of overloaded function “std::min” matches the argument list
argument types are: (, long)

D:/detectron2/detectron2/layers/csrc/ROIAlign/ROIAlign_cuda.cu(394): error: no instance of function template “at::cuda::ATenCeilDiv” matches the argument list
argument types are: (int64_t, long)

D:/detectron2/detectron2/layers/csrc/ROIAlign/ROIAlign_cuda.cu(394): error: no instance of overloaded function “std::min” matches the argument list
argument types are: (, long)
解决办法:
全改成int64_t
参考链接:https://github.com/open-mmlab/mmdetection/issues/1088
2)ValueError: path ‘/home/lc/detectron2/detectron2/layers/csrc/cuda_version.cu’ cannot be absolute
删除掉对应的egg.info 文件夹
然后用以下指令重新编译:
python setup.py build --force develop
参考链接:
https://stackoverflow.com/questions/43080203/path-s-cannot-be-absolute-pathname
##2.跑代码时遇见:
1)RuntimeError: Not compiled with GPU support (ROIAlign_forward at E:\detectron2\detectron2\layers\csrc\ROIAlign/ROIAlign.h:73)
(no backtrace available)

先检查cuda版本和pytorch版本是否对应,然后重新安装
python setup.py build --force develop

最后:
1.感谢上述链接的作者的无私分享内容。
2.帖子内容可能不全,欢迎大家补充,交流。

  • 9
    点赞
  • 26
    收藏
    觉得还不错? 一键收藏
  • 40
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 40
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值