AI开发平台ModelArtsAI开发平台ModelArts

文档首页> AI开发平台ModelArts> 常见问题> 模型管理> 导入模型> 导入模型时安装包存在前后依赖关系,模型配置文件应该如何说明?
更新时间:2021/03/18 GMT+08:00
分享

导入模型时安装包存在前后依赖关系,模型配置文件应该如何说明?

问题描述

从OBS中或者从容器镜像中导入模型时,开发者需要编写模型配置文件。模型配置文件描述模型用途、模型计算框架、模型精度、推理代码依赖包以及模型对外API接口。配置文件为JSON格式。配置文件中的“dependencies”表示模型推理代码需要依赖的包。模型开发者需要提供包名、安装方式、版本约束,详细参数见参数说明表。“dependencies”参数填入“dependency”结构数组。当导入模型时安装包存在前后依赖关系,模型配置文件中的“dependencies”参数应该如何说明?

解决方案

当安装包存在前后依赖关系时,模型配置文件的“dependencies”支持多个“dependency”结构数组以list形式填入。

list形式的“dependencies”有顺序要求,例如您想安装“mmcv-full”“mmcv-full”,安装之前需要完成“Cython”“pytest-runner”“pytest”的安装。在配置文件中,采用list形式填入安装包时,您需要把“Cython”“pytest-runner”“pytest”写在“mmcv-full”结构数组的前面。

示例如下:

"dependencies": [
    {
    "installer": "pip",
    "packages": [
        {
            "package_name": "Cython"
        },
        {
            "package_name": "pytest-runner"
        },
        {
            "package_name": "pytest"
        }]
    },

    {
    "installer": "pip",
    "packages": [
        {
            "restraint": "ATLEAST",
            "package_version": "5.0.0",
            "package_name": "Pillow"
        },
        {
            "restraint": "ATLEAST",
            "package_version": "1.4.0",
            "package_name": "torch"
        },
        {
            "restraint": "ATLEAST",
            "package_version": "1.19.1",
            "package_name": "numpy"
        },
        {
            "restraint": "ATLEAST",
            "package_version": "1.2.0",
            "package_name": "mmcv-full"
        }]
    }
]

分享:

导入模型 所有常见问题

more