关于win10系统下Detectron2框架搭建的一些心得

关于Detectron2框架搭建的一些心得

参考链接:https://www.bilibili.com/video/BV1jZ4y1W7Nb

感谢up主的详细讲解,我在网上搜索了大量的关于detectron2框架的安装教程,然后再根据up主的详细讲解,最终完成了detectron2的安装。

首先说一下自己遇到的问题:

作为一名新手,第一次配深度学习框架的环境,着实有点艰难。
detectron2的框架搭建可分为三个部分,首先是NVIDIA驱动的安装以及cuda版本和cudnn版本的安装,这个网上的教程都一大堆。我最终下载下来这三个文件
在这里插入图片描述在这里插入图片描述
依次安装驱动、cuda、然后将cudnn压缩包解压,并将cudnn解压下来的三个文件夹复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1中

然后将环境路径设置好

最终呈现的效果就是打开cmd,输入 nvcc -V 可以出现以下图片:在这里插入图片描述
如图所示,我用的是cuda10.1.243

这个安装完成后,就可以在anaconda中创建环境,并且按照视频步骤,一步一步安装了。(关于detectron2安装资源,我是通过up主的公众号获取的

首先需要安装就是pytorch,我这里使用的是up主所说的版本,pytoch=1.4,cuda=10.1,这个在官网中都有,不过下载起来可能比较慢,(我当时确实下载了很久,失败了很多次才下载完成。)

然后就是安装所需的依赖库了,如cython、fvcore、matplotlib等,

这里推荐一个pip源,下载速度飞快:-ihttp://pypi.douban.com/simple --trusted-host pypi.douban.com

接着就是cocotools安装,这个也不多赘述,按照步骤安装,一般不会出问题。

然后修改up主所说的几个地方,一直到最后detectron2的安装
最终安装完成,可以用conda list看到
在这里插入图片描述
detectron2已经安装完成。

然后进入模型所在文件夹,对框架进行测试,输入以下命令:
python demo/demo.py --config-file configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml --input 1.jpg --opts MODEL.WEIGHTS model_final_a54504.pkl
如下所示:
在这里插入图片描述

在这里插入图片描述
这说明,框架搭建成功!

我遇到的问题:

1、关于CUDA erro: no ernel image is available for excution on the device 的错误。

这个错误是安装完成detectron2之后会出现的,我当时安装完以为完事了,结果测试时报错了上面的错误。
在视频下方的评论中有提到,大体意思就是需要在输入搭建detectron2框架的命令行时,加入–force
原先代码是:python setup.py build develop
应更改为:python setup.py build --force develop

但此时我这出现了新的错误:error: command F:\Anaconda3\anaconda\envs\tensorflow2\Library\bin\nvcc failed: No such file or directory

找了好久才知道是我电脑之前安装过tensorflow,后面环境被我删了,CUDA路径没改,直接找不到路径了。(若是根据我上面所说的设置CUDA路径不会有这个问题)

修改完成后,还是报错:C:/Program Files (x86)/Microsoft Visual Studio/2019/Community/VC/Tools/MSVC/14.29.30037/include\yvals_core.h(531): fatal error C1189: #error: STL1002: Unexpected compiler version, expected CUDA 10.1 Update 2 or newer.

当时也问了UP主,说是CUDA没安装好。

我一直以为是vs版本的原因,之后痛定思痛,重新将CUDA和cudnn安装了一遍,就是上述的安装过程。在进行测试后,运行成功。

原因:在Anaconda中,我的环境安装的为pytorch=1.4,cudatoolkit为10.1,但其实他的具体版本为10.1.243

而在电脑中,我没有重新安装CUDA和cudnn前,cuda版本为10.1.105,

当时的我没有想太多,没想到就是后面这个版本号没有对应上,最终安装没成功,果然是细节决定成败。
在这里插入图片描述

全当是自己安装的笔记了。

再次感谢up主的详细讲解!

有错误的话,麻烦大佬帮忙指正!!万分感谢!!

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值