MMCV2.0新手安装避坑

最近想试试MMCV2.0,之前是已经有1.x的环境了,然后发一些需要注意的点。

环境是自己的本地环境,window11+GTX4070,这里我是按照官方教程安装的:开始你的第一步 — MMDetection 3.0.0 文档

1.pytorch和cuda版本必须匹配,这我就不赘述了,详情查询pytorch安装攻略,注意,不要按照官方文档的命令去安装,会自动安装cpu版本,建议是安装一些成熟的版本,目前不要安装pytorch2.0.0以及上环境。(因为pytorch和NVIDIA会有包下载不下来卡住)

2.使用mim安装MMCV时,会自动识别pytorch版本,也就是说,只要安装的是gpu版本的pytorch,并且确认显卡availble,mim会自动安装对应版本的MMCV2.0gpu版

3.在mmdetection验证安装步骤,如果是mim安装的,可以打开mmdetection官方代码仓库下载image_demo代码,命令行cd到下载的模型参数目录下,在运行代码,即可得到生成结果,不必再调用python解释器运行。(因为原教程给的是生成标注list,可视化效果不好。)

python demo/image_demo.py demo/demo.jpg rtmdet_tiny_8xb32-300e_coco.py --weights rtmdet_tiny_8xb32-300e_coco_20220902_112414-78e30dcc.pth --device cpu

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值