更新时间:2024-02-28 GMT+08:00

Notebook基础镜像列表

ModelArts开发环境提供Docker容器镜像,可作为预构建容器运行。预置镜像里面包含PyTorch,Tensorflow,MindSpore等常用AI引擎框架,镜像命名以AI引擎为主,并且每个镜像里面都预置了很多常用包,用户可以直接使用而无需重新安装。

表1 X86预置镜像列表

引擎类型

镜像名称

PyTorch

pytorch1.8-cuda10.2-cudnn7-ubuntu18.04

pytorch1.10-cuda10.2-cudnn7-ubuntu18.04

pytorch1.4-cuda10.1-cudnn7-ubuntu18.04

Tensorflow

tensorflow2.1-cuda10.1-cudnn7-ubuntu18.04

tensorflow1.13-cuda10.0-cudnn7-ubuntu18.04

MindSpore

mindspore1.7.0-cuda10.1-py3.7-ubuntu18.04

mindspore1.7.0-py3.7-ubuntu18.04

mindspore1.2.0-cuda10.1-cudnn7-ubuntu18.04

mindspore1.2.0-openmpi2.1.1-ubuntu18.04

无AI引擎(专用于自定义镜像的基础镜像)

conda3-cuda10.2-cudnn7-ubuntu18.04

conda3-ubuntu18.04