AI开发平台MODELARTS-自定义模型规范:自定义模型使用的预置镜像

时间:2024-04-11 20:22:53

自定义模型使用的预置镜像

AI Gallery提供了PyTorch基础镜像,镜像里已经安装好了运行任务所需的软件,供自定义模型直接使用,快速进行训练、推理。预置镜像的版本信息请参见表3

表3 AI Gallery预置镜像列表

引擎类型

资源类型

版本名称

PyTorch

NPU

pytorch_2.0.1-cann_6.3.2-py_3.9-euler_2.10.7-aarch64

GPU

pytorch_2.0.0-cuda_11.7-py_3.9.11-ubuntu_20.04-x86_64

support.huaweicloud.com/aimarket-modelarts/ma_gallery_0055.html