Updated on 2024-08-14 GMT+08:00

Notebook Base Image List

ModelArts DevEnviron provides Docker container images, which can run as preset containers. Certain preset images are built on common AI engine frameworks such as PyTorch, TensorFlow, and MindSpore. These images are named using the AI engines. Additionally, many common packages are preset in these images, relieving you from the package installation.

Table 1 Preset x86 images

AI Engine

Image

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

No AI engine (base images dedicated for image customization)

conda3-cuda10.2-cudnn7-ubuntu18.04

conda3-ubuntu18.04