本文主要是介绍【mmdetection安装】在线安装和离线安装,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
mmdetection安装
- 安装版本
- 在线安装版
- 离线安装版
- 参考
记录mmdetection安装过程,网络环境好久就选择在线安装,记录离线安装是因为超算网络环境太恶劣…
安装版本
版本 | |
---|---|
python | 3.10 |
pytorch | 2.1.0(cuda11.8) |
mmcv | 2.1.0 |
mmdetection | git下载安装即可 |
在线安装版
虚拟环境:
conda create -n mmdet python=3.10
conda activate mmdet
pytorch:使用2.3.0折腾了好一会,然后退回到了2.1.0
pytorch 2.1.0
# cuda11.8
conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=11.8 -c pytorch -c nvidia
使用pip安装mmcv:mmcv2.1.0
pip install -U openmim
pip install mmengine
pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.1/index.html
安装mmdetection:
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -v -e .
验证安装:
from mmdet.apis import DetInferencer# 初始化模型
inferencer = DetInferencer('rtmdet_tiny_8xb32-300e_coco')# 推理示例图片
inferencer('../demo/demo.jpg', out_dir='outputs/', no_save_pred=False)
离线安装版
虚拟环境:
conda create -n mmdet python=3.10
conda activate mmdet
pytorch:
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
# CUDA 12.1
pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121
安装cuda11.8版本,去官网指定地址https://download.pytorch.org/whl/cu118下载.whl安装文件:torch==2.1.0
、torchvision==0.16.0
、torchaudio==2.1.0
注意安装gpu版本:
torch-2.1.0+cu118-cp310-cp310-linux_x86_64.whl
torchvision-0.16.0+cu118-cp310-cp310-linux_x86_64.whl
torchaudio-2.1.0+cu118-cp310-cp310-linux_x86_64.whl
安装 pip install torch*
使用pip安装mmcv:mmcv2.1.0
pip install -U openmim
pip install mmengine
pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.1/index.html
去官网指定下载地址https://download.openmmlab.com/mmcv/dist/cu118/torch2.1/index.html下载对应版本mmcv:
mmcv-2.1.0-cp310-cp310-manylinux1_x86_64.whl
安装:pip install mmcv-2.1.0-cp310-cp310-manylinux1_x86_64.whl
安装mmdetection:
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -v -e .
验证安装:
from mmdet.apis import DetInferencer# 初始化模型
inferencer = DetInferencer('rtmdet_tiny_8xb32-300e_coco')# 推理示例图片
inferencer('../demo/demo.jpg', out_dir='outputs/', no_save_pred=False)
参考
INSTALLATION - MMCV
GET STARTED - MMDetection
Cold_Chair的博客 mmdetection 安装(不支持 pytorch 2.2.x)
这篇关于【mmdetection安装】在线安装和离线安装的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!