时间: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的区别;
2、安装vs2019;
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
5、安装cocoapi;
下载:https://github.com/philferriere/cocoapi,
cd到cocoapi/PythonAPI,输入python setup.py build_ext install
安装Cython后重新输入python setup.py build_ext install
6、下载fvcore;
下载:https://github.com/facebookresearch/fvcore
cd到vcore,输入python setup.py build --force develop
7、下载detectron2(0.4);
下载:https://github.com/facebookresearch/detectron2/tree/v0.4
cd 到detectron2,执行python setup.py build --force develop,一堆错误;
8、修改代码,包括如下这四个文件,如果下载有多个版本时不要选错了;
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())
2:找到ROIAlignRotated_cuda.cu、deform_conv_cuda_kernel.cu,把里面的ceil改为ceilf、floor改为floorf
3:ir.h的1347行,static constexpr Symbol Kind = ::c10::prim::profile_optional,
注释掉
重新执行python setup.py build --force develop
9、成功;
参考链接:
https://zhuanlan.zhihu.com/p/363295722