OpenMMLab环境配置的保姆级流程

1.环境

确认当前安装的cuda版本

Installation — mmcv 2.2.0 文档

在这个文档查看你能兼容的mmcv版本,如果不行,重新装一个cuda、
多个cuda教程见Windows10下多版本CUDA的安装与切换 超详细教程_cuda版本-CSDN博客

mmcv不能包含2.1.0

推荐用cuda11.8,然后安装pytorch

11.8的cuda的torch安装命令
conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia

接下来按照依赖 — MMYOLO 0.6.0 文档

一步一步执行就行,前期的环境确认就会减少很多坑

后续我按照官方流程全复制粘贴 直到最后验证demo有问题,就是他给的命令复制过来用不了我们用如下命令验证:

python demo/image_demo.py demo/demo.jpg yolov5_s-v61_syncbn_fast_8xb16-300e_coco.py yolov5_s-v61_syncbn_fast_8xb16-300e_coco_20220918_084700-86e02187.pth --device cpu --out-dir result.jpg

把这个代码 输入 就是输出一个result.jpg的文件夹,里面就有这个图哦,然后就开启mmyolo的学习之旅咯

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值