MMCV2.0新手安装避坑

最新推荐文章于 2024-08-14 06:13:01 发布

好数据胜过好算法

最新推荐文章于 2024-08-14 06:13:01 发布

阅读量1.2k

点赞数 1

文章标签：计算机视觉

本文链接：https://blog.csdn.net/qq_43176280/article/details/131002929

版权

在Windows11环境下，升级到MMCV2.0需要注意pytorch和cuda版本匹配，避免安装CPU版本。使用mim安装MMCV会自动适配gpu版本。验证安装成功可直接运行MMDetection的image_demo，无需额外配置。

摘要由CSDN通过智能技术生成

最近想试试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