mmdetection 安装(不支持 pytorch 2.2.x)

参考:
https://mmdetection.readthedocs.io/zh-cn/latest/get_started.html

有点坑的是安装 mmcv 时去 https://mmcv.readthedocs.io/en/latest/get_started/installation.html 上面一查,
发现不支持 pytorch 2.2.x(截止 2024.4.5)
按教程的方法装不会报错,但是就是跑不起来程序,真傻逼。


复制conda create --name openmmlab python=3.10 -y
conda activate openmmlab

先上:
https://pytorch.org/get-started/previous-versions/
装个 pytorch 2.1.0,我本机 CUDA 11.8,所以是:

复制pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118
复制pip install -U openmim

装这个 mmengine 随便装。

复制pip install mmengine

然后去上面找:
https://mmcv.readthedocs.io/en/latest/get_started/installation.html
指令为:

复制pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.1/index.html

最后:

复制git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -v -e .
# "-v" 指详细说明,或更多的输出
# "-e" 表示在可编辑模式下安装项目,因此对代码所做的任何本地修改都会生效,从而无需重新安装。

然后跑跑这个代码:

复制from mmdet.apis import DetInferencer

# 初始化模型
inferencer = DetInferencer('rtmdet_tiny_8xb32-300e_coco')

# 推理示例图片
inferencer('../mmdetection/demo/demo.jpg', out_dir='outputs/', no_save_pred=False)

能正常输出就没有问题了

posted @   Cold_Chair  阅读(197)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· winform 绘制太阳,地球,月球 运作规律
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
历史上的今天:
2020-04-05 「THUSCH 2017」换桌(zkw费用流)—对几种费用流算法的总结
2020-04-05 「THUSCH 2017」杜老师(找性质+线性基)
2020-04-05 「THUSCH 2017」巧克力(斯坦纳树dp+随机化+中位数二分)
2020-04-05 [ZJOI2011]最小割(最小割树模板)
点击右上角即可分享
微信分享提示