win10安装detectron2,亲测有效

时间:2021.5.7

环境:win10、detectron2(0.4)、python3.6.13、pytorch1.7.1、torchvision0.8.2、cuda10.1、cudnn7.6、vs2019

1、安装cuda及cudnn;结果以nvcc -V结果为准,与nvidia-smi的区别;

win10安装detectron2,亲测有效_第1张图片

2、安装vs2019;

win10安装detectron2,亲测有效_第2张图片

 

3、安装Anaconda,打开Anaconda Prompt

conda create -n detectron2 python==3.6.13

conda activate detectron2

4、安装pytorch,找到pytorch1.7.1,先确保安装正确;

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

win10安装detectron2,亲测有效_第3张图片

5、安装cocoapi;

下载:https://github.com/philferriere/cocoapi,

cd到cocoapi/PythonAPI,输入python setup.py build_ext install
win10安装detectron2,亲测有效_第4张图片

安装Cython后重新输入python setup.py build_ext install

6、下载fvcore;

下载:https://github.com/facebookresearch/fvcore

cd到vcore,输入python setup.py build --force develop
win10安装detectron2,亲测有效_第5张图片

7、下载detectron2(0.4);

下载:https://github.com/facebookresearch/detectron2/tree/v0.4

cd 到detectron2,执行python setup.py build --force develop,一堆错误;
win10安装detectron2,亲测有效_第6张图片

8、修改代码,包括如下这四个文件,如果下载有多个版本时不要选错了;

win10安装detectron2,亲测有效_第7张图片

1:cpp_extension.py中大约280行,

match = re.search(r'(\d+)\.(\d+)\.(\d+)', compiler_info.decode().strip())修改为

match = re.search(r'(\d+)\.(\d+)\.(\d+)', compiler_info.decode(' gbk').strip())

win10安装detectron2,亲测有效_第8张图片

2:找到ROIAlignRotated_cuda.cu、deform_conv_cuda_kernel.cu,把里面的ceil改为ceilf、floor改为floorf

win10安装detectron2,亲测有效_第9张图片

3:ir.h的1347行,static constexpr Symbol Kind = ::c10::prim::profile_optional,注释掉

重新执行python setup.py build --force develop

9、成功;

win10安装detectron2,亲测有效_第10张图片

参考链接:

https://zhuanlan.zhihu.com/p/363295722

https://blog.csdn.net/iamleifeng/article/details/103877581

https://blog.csdn.net/tanmx219/article/details/100829920

你可能感兴趣的:(深度学习,detectron2,pytorch,cuda10.1,深度学习)