Detectron2v4环境安装、
第一次装detectron2这框架一堆bug安装了好几天,最近电脑重装系统,太久安装忘记了,又安装了一天,于是这次就顺便写个博客记录一下,希望大家少走坑,可以参考一下。
环境准备
首先安装需要确定一下自己的电脑有没有安装anaconda以及cuda、cudnn(csdn中有比较详细的介绍,这里不过详细阐述)。
本人的设备以及环境环境介绍
GTX960+cuda10.1+cudnn8.0.5+py3.7+torch1.8
安装VS2019
这里仅安装c++模块即可,如下图所示。
其次找到cl.exe的所处位置,配置环境变量。
本人安装的路径为 D:\vs2019\vs\VC\Tools\MSVC\14.29.30133\bin\Hostx86\x86 将其添加到Path下
在cmd中输入cl是否有以下信息,则安装成功。
最后还需要在该命令行中输入
set DISTUTILS_USE_SDK=1
VS2019安装就告一段落。
python 3.7
首先创建python=3.7的虚拟环境
conda create -n detectron2 python=3.7
conda activate detectron2
torch 1.8.1
随后安装torch
这里需要根据大家机器的cuda以及torch的版本去pytorch官网获取安装命令
pip install torch1.8.1+cu101 torchvision0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
此外如果下载速度比较慢的情况,可以考虑一下换源。
cocoapi安装
cd 到 E:\pycocotools\dt2v4\cocoapi-master\cocoapi-master\PythonAPI
输入pip install -e.
fvcore安装
cd 到 E:\pycocotools\dt2v4\fvcore-master\fvcore-master
python setup.py build --force develop
ninja安装
pip install ninja
最后一步
cd到E:\pycocotools\dt2v4\detectron2-0.4
python setup.py build
python setup.py install
cocoapi、fvcore、detectron2v4 文件在下链接获取
链接:https://share.weiyun.com/SAoCCcvN 密码:7jiw8c