mmdeploy环境配置
时间: 2023-08-19 16:10:52 浏览: 169
为了配置mmdeploy环境,你可以按照以下步骤进行操作:
1. 创建一个名为mmdeploy的conda环境,并指定Python版本为3.8:
```
conda create --name mmdeploy python=3.8 -y
```
2. 激活mmdeploy环境:
```
conda activate mmdeploy
```
3. 安装PyTorch,可以根据你的需求选择特定的版本。以下是一个示例命令,你可以根据需要进行调整:
```
conda install pytorch=={pytorch_version} torchvision=={torchvision_version} cudatoolkit={cudatoolkit_version} -c pytorch -c conda-forge
```
4. 克隆mmdeploy仓库并安装依赖:
```
git clone --recursive https://github.com/open-mmlab/mmdeploy.git
cd mmdeploy
python -m pip install -r requirements/runtime.txt
```
5. 编译Model Converter自定义算子库:
```
cd mmdeploy
mkdir -p build && cd build
cmake -DCMAKE_CXX_COMPILER=g -DMMDEPLOY_TARGET_BACKENDS=trt -DONNXRUNTIME_DIR=${ONNXRUNTIME_DIR} ..
make -j8 && make install
```
6. 安装Model Converter:
```
cd mmdeploy
pip install -e .
```
通过按照以上步骤进行操作,你就可以成功配置mmdeploy环境了。请注意,其中的一些命令可能需要根据你的具体环境和需求进行调整。
阅读全文