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

你可能感兴趣的:(计算机视觉)