1.环境
确认当前安装的cuda版本
在这个文档查看你能兼容的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的学习之旅咯