跳到主要内容

依赖管理

Serving Predictor 中有介绍各个模式下的预装依赖。如果需要安装额外的依赖,可以按照以下的方式做配置,相应的依赖会在部署服务启动之前进行安装。

PyPI 依赖

在模型部署文件夹的根目录下如果存在名为 requirements.txt 的文件,OpenBayes 模型部署在启动前会执行 pip install -r requirements.txt 命令, 安装其中声明的 PyPI 库。

Conda 依赖

OpenBayes 模型部署也同样支持 Conda 包的安装。在部署服务启动之前会在部署目录的根目录下查找名为 conda-packages.txt 的文件,其文件格式遵循

[channel::]package[=version[=buildid]]

这里有一个样例:

conda-packages.txt
conda-forge::rdkit
conda-forge::pygpu

如果 requirements.txtconda-packages.txt 同时存在,那么会首先安装 conda-packages.txt 里的依赖,然后再安装 requirements.txt 中的依赖。

其他依赖

对于非 Conda 和 PyPI 的依赖,可以在根目录提供一个名为 dependencies.sh 的文件,在模型部署启动时它会被 bash 执行,并且其执行会早于 requirements.txtconda-packages.txt 依赖的安装。

这里提供一个安装 tree 应用的例子:

dependencies.sh:

apt update && apt install tree -y