MMDeploy 学習ノート (2): MMDeploy の環境設定
MMDeploy 環境設定
MMCVやMMDetectionなどの基本的な環境をインストールしてから、MMDeployの環境設定を行う
MMDeploy は、OpenMMLab によって開始されたオープンソースのモデル展開アルゴリズム ライブラリであり、各アルゴリズム ライブラリに統一された展開操作を提供することを目的としています。MMDeploy に基づいて、開発者は指定されたハードウェアに必要な SDK をトレーニング アルゴリズム ライブラリから簡単に生成できるため、適応にかかる時間を大幅に節約できます。
MMDeploy ホームページ: https://github.com/open-mmlab/mmdeploy
インストール
- MMDeploy の最新バージョンをダウンロードする
git clone https://github.com/open-mmlab/mmdeploy.git --recursive
cd mmdeploy
- 元のソースをバックアップし、Tsinghua ソースに置き換えます (システム自体が既に Tsinghua ソースである場合は、この手順をスキップしてください)。
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
- MMDeploy をコンパイルしてインストールします (約 10 分かかります)。
python tools/scripts/build_ubuntu_x64_ort.py
# 上面的脚本,包括了以下几个子步骤:
# 1、配置工具链
# 2、下载并安装推理引擎 onnxruntime,和其他的依赖项
# 3、编译并安装 mmdeploy 中的 onnxruntime 自定义算子和 mmdeploy inference SDK
# 4、安装 mmdeploy python 包
- 環境変数の構成
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'
エラーが報告されなければ、インストールは成功です。