maskrcnn-benchmark安装记录

前言

因为学校里有个工程实训的课程,正好老师就说让跑一下maskrcnn benchmark看看识别效果怎么样。然后我就去下了,这里来记录一下安装过程中遇到的困难和问题。本来是让试用一下seg_every_thing那个框架的,结果因为本人显存不够大所以并没有运行成功。最后下了facebookresearch的那个改良版的(?)。链接在这里:

https://github.com/facebookresearch/maskrcnn-benchmark

先从成功的配置环境开始讲,遇到的问题都在“试错”标题下。

配置

系统:Pop_os! 19.04 https://system76.com/pop

硬件:GTX 970M

CUDA10.0、cudnn7.5.1,gcc6.5、pytorch1.1.0,pytorch-nightly

选择pop_os这个系统的原因是它是基于ubuntu的,而且自己下载好了nvidia驱动(不用再自己费劲巴拉的安装),而且我是windows10下一直都无法编译apex那个包(后面会提到),初步估计是因为我vs的版本太高了(2017的15.9.11),导致cuda环境与之不匹配(windows下因为要跑maskrcnn,所以装的是cuda9+cudnn7)。

但是要注意的是,这个系统选好了安装盘就直接在那个盘里安装了,如果装双系统的话我也不清楚,因为我自己是装在硬盘下的。这个系统确实挺友好的,反正我挺喜欢~就是有些许一些小bug,比如说没法切换大小写(我还没找到解决方法)。而且它出了18.04LTS版本的。再往下的版本是没有的,反正我翻遍了论坛和官网,pop_os是本来是有一个17.04版的,后来不提供了。

安装过程

一、安装cuda10.0和cudnn7.5.1

首先下载cuda10.0和cudnn7.5.1。这里借鉴了这个博主的安装过程:

https://blog.csdn.net/wanzhen4330/article/details/81699769

非常简单,下好了cuda10.0的.run文件,安装时用

sudo sh (你下的版本).run --override

就可以了。至于为啥要

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值