华为云AI开发平台ModelArts算法工程环境管理_云淘科技

9 月 26, 2023

157 0

EnvManager是针对于AI开发套件工程化管理的抽象，主要用于创建算法工程、查看算法/模型/数据集等资产、安装依赖环境、切换工作路径等。

from modelarts.algo_kits import EnvManager
env = EnvManager()
env.init_env(install_path=None, project_name=None)   # 初始化工程

查看模型信息，可以传入filter_keyword进行信息过滤。

env.show_asset(mode="model", filter_keyword=["coco", "fcos"])

查看数据集信息。

env.show_asset("dataset")

查看套件信息。

env.show_asset("algorithm")

安装算法套件，Notebook实例启动后，首次安装套件后可能需要重新启动kernel以使用更新后的依赖包。

env.install(mode="algorithm", asset_name="mmdetection", version="2.17.0")

安装数据集。

env.install(mode="dataset", asset_name="coco2017_sample")

安装预训练模型。

env.install('model', 'mmdetection:fcos/fcos_r50_caffe_fpn_gn-head_1x_coco')

切换工作路径。

env.change_work_path('your_work_path')

安装套件依赖环境。

env.change_work_path('your_work_path')

安装预编译的mmcv-full，减少编译mmcv-full的时间。

env.install_mmcv_full(version="1.3.9",
                      torch="1.8.0",
                      cuda="10.2",
                      python="3.7.0")

父主题： 通过Python API使用算法套件

内容没看懂？不太想学习？想快速解决？有偿解决：联系专家

分类