安装detectron2
自述经历
在windows下配置这些环境什么的 报的错都是特别离谱的???网上搜也搜不到! 本人还是建议有条件的还是使用linux系统吧!!!!!
安装条件
需要配置好相应版本的NVIDIA驱动、CUDA、Pytorch、visual studio,visual stdio2019是为了配置gcc g++的。
配置pytorch的链接link.
配置CUDA的链接link.
本人安装的是最新的detectron2版本,使用的pytorch、python、cuda、visual studio版本如下
本人显卡为:GTX 3060
NVIDIA 驱动:511.70
CUDA :11.1
pytorch:1.8.0
cudatoolkit:11.1.1
visual studio :2019
安装步骤
新建anaconda环境
link.
下载opencv-python
在环境命令行中输入
pip install opencv-python
拉取detectron2包
给出detectron2在github上的链接link.
https://github.com/facebookresearch/detectron2/tree/main
点击右上角的code,可以下载压缩包,也可以使用git命令进行在命令行中拉取。
git命令如下
git clone https://github.com/facebookresearch/detectron2.git
下载到相应的文件夹中,然后解压 ,如果是git方式就不需要解压了。
然后在你所建的conda环境中,cd进detectron2的文件夹中,
然后
构建
python -m pip install -e detectron2
自动会安装好相应所需要的软件包,等待一会即可安装好!
然后,在detectron2文件夹中存放入一个测试的图片,命名为1.jpg
然后进入detectron2 文件夹,继续输入
cd detectron2
python demo/demo.py --config-file configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml --input 1.jpg --output result --opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl
然后使用自动会下载模型,进行预测。
错误相应集合,之后更新
只要自己的上述都安装的没问题了,就可以顺顺利利。
错误类型
visual studio2019
环境没有添加 ,也有可能是之前版本卸载没有卸载干净 报错 出现 cl.exe什么错误!!!
检查环境
版本没卸载干净解决办法:打开anaconda命令终端,注意不是cmd终端!!切换到相应虚拟环境下
输入(看自己下载的是社区版本还是个人版本)
call "C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Auxiliary\Build\vcvars64.bat"
SET DISTUTILS_USE_SDK=1
个人版本就是
call "C:\Program Files (x86)\Microsoft Visual Studio\2019\Professional\VC\Auxiliary\Build\vcvars64.bat"
SET DISTUTILS_USE_SDK=1
pytorch 与cuda版本问题
上网搜索
最后参考链接,他是用的0.2版本的deterctron2 ,现在最新的是0.6以上了。
链接: link.