MMDeploy学习笔记(二):MMDeploy环境配置

MMDeploy学习笔记(二):MMDeploy环境配置

MMDeploy环境配置

首先安装MMCV、MMDetection等基础环境,然后进行MMDeploy的环境配置

MMDeploy是OpenMMLab推出的开源模型部署算法库,旨在为各算法库提供统一的部署操作。基于MMDeploy,开发者可以轻松从训练算法库生成指定硬件所需SDK,省去大量适配时间,此博客便介绍了MMDeploy如何进行环境的配置。

MMDeploy主页:https://github.com/open-mmlab/mmdeploy

安装

  1. 下载最新版MMDeploy
git clone https://github.com/open-mmlab/mmdeploy.git --recursive
cd mmdeploy
  1. 备份原始源,并替换为清华源(若系统本身已经是清华源,请跳过此步骤)
sudo cp /etc/apt/sources.list /etc/apt/sources.list.backup
sudo sed -i "s@http://.*archive.ubuntu.com@https://mirrors.tuna.tsinghua.edu.cn@g" /etc/apt/sources.list
sudo sed -i "s@http://.*security.ubuntu.com@https://mirrors.tuna.tsinghua.edu.cn@g" /etc/apt/sources.list
sudo apt-get update
  1. 编译并安装 MMDeploy(耗时大约十分钟)
python tools/scripts/build_ubuntu_x64_ort.py
# 上面的脚本,包括了以下几个子步骤:
# 1、配置工具链
# 2、下载并安装推理引擎 onnxruntime,和其他的依赖项
# 3、编译并安装 mmdeploy 中的 onnxruntime 自定义算子和 mmdeploy inference SDK
# 4、安装 mmdeploy python 包
  1. 配置环境变量
export PYTHONPATH=$(pwd)/build/lib:$PYTHONPATH
export LD_LIBRARY_PATH=$(pwd)/build/lib:$(pwd)/../mmdeploy-dep/onnxruntime-linux-x64-1.8.1/lib:$LD_LIBRARY_PATH

验证安装成功

在命令行中输入如下指令:

python -c 'import mmdeploy'
python -c 'from mmdeploy_python import Detector'

若无报错,则说明安装成功。

猜你喜欢

转载自blog.csdn.net/weixin_43603658/article/details/129714765
今日推荐