GPU安装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.

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值