ubuntu配置mmdetection 的CPU版本

由于大多数的服务器没有GPU,我们想要在上面跑openmmlab项目中的mmdetection需要装CPU版本的,由于官方的感觉有点模糊,所以自己搜索资料跑通了一个[mmdetection连接]
(https://github.com/open-mmlab/mmdetection)、
当然mmlab其他项目应该也是类似的

1.首先老样子,创建环境

conda create -n mmlab python=3.6 -y
conda activate open-mmlab

2.然后,安装torch的cpu版本

pip install torch==1.5.0 torchvision==0.6.0

这个可能有点慢,分享一手whl

3.接着,安装mmcv-full

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.5.0/index.html

注意最后的torch版本要根自己的版本对应

4.再接着安装requirements和setup.py

pip install -r requirements/build.txt
cd {mmdetection存在路径}
python setup.py develop

5.最后,跑代码

python demo/image_demo.py demo/demo.jpg configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.py weights/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_20200130-f513f705.pth --device cpu

第一个

demo/image_demo.py #这个是识别调用接口

第二个

demo/demo.jpg #识别的图片

第三个

configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.py#调用的是哪个算法模块

第四个

weights/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_20200130-f513f705.pth#这个是权重文件的位置,这个weights的文件是我自己创建的

权重文件可以在configs文件夹下某一个如faster_rcnn文件夹中的
README.md文件中找到,一定要找对应的

第五个

--device cpu#这个就是运行的设备,是cpu

还有就是最近新出的一个involuntion算子想用cpu跑通用在项目中不过貌似必须要cuda

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值