全部產品
Search

搜尋本產品

    Platform For AI：PAI官方鏡像

    文件中心

    Platform For AI：PAI官方鏡像

    更新時間：Jul 13, 2024

    阿里雲PAI為您提供了基於不同架構、不同CUDA版本的官方鏡像，您可以在使用DLC、EAS、DSW時直接選擇適用的鏡像，即可快速基於官方鏡像構建AI開發環境。本文為您介紹PAI預置的官方鏡像能力和核心鏡像列表。

    認識官方鏡像

    阿里雲PAI為您提供的官方鏡像遵循一定的命名規則，您可以通過鏡像名稱瞭解到此鏡像的基本資料。通常鏡像命名會包含以下幾個固定欄位，建議您添加自訂鏡像時使用統一的命名規則。

    官方鏡像名稱樣本

    鏡像名稱解讀

    各個產品支援使用的鏡像類型

    tensorflow:2.11-gpu-py39-cu112-ubuntu20.04

    • tensorflow:2.11：表示訓練架構為2.11版本的TensorFlow。

    • gpu：表示適用GPU類型的機型。

    • py39：表示開發語言為3.9版本的Python。

    • cu112：表示支援的CUDA版本為112版本。

    • ubuntu20.04：表示支援的作業系統為20.04版本的Ubuntu作業系統。

    官方鏡像適用的子產品不同：

    • -training類型的鏡像適用於DLC。

    • -inference類型的鏡像適用於EAS。

    • -develop類型的鏡像適用於DSW。

    deeprec-develop:2302-tensorflow1.15-cpu-py36-ubuntu18.04

    • deeprec-develop:2302-tensorflow1.15：表示訓練架構為1.15版本的TensorFlow和2302版本的DeepRec。

    • cpu：表示適用於CPU類型的機型。

    • py36：表示開發語言為3.6版本的Python。

    • ubuntu18.04：表示支援的作業系統為18.04版本的Ubuntu作業系統。

    官方鏡像能力

    阿里雲PAI為您提供了基於不同機器學習架構的官方鏡像，您可以在PAI控制台的鏡像AI資產頁面中查看所有官方鏡像列表，以下為您介紹基於TensorFlow、PyTorch等主流架構的官方鏡像資訊。

    TensorFlow

    架構版本

    CUDA版本（僅GPU機型）

    作業系統

    • TensorFlow2.6

    • TensorFlow2.3

    • TensorFlow2.21

    • TensorFlow2.11

    • TensorFlow1.15，TensorFlow1.15.5

    • TensorFlow1.12

    • CUDA 114

    • CUDA 113

    • CUDA 112

    • CUDA 101

    • CUDA 100

    • Ubuntu 20.04

    • Ubuntu 18.04

    TensorFlowServing

    架構版本

    CUDA版本（僅GPU機型）

    作業系統

    • TensorFlowServing2.11.1

    • TensorFlowServing1.15.0

    • CUDA 112

    • CUDA 100

    • Ubuntu 20.04

    • Ubuntu 18.04

    • Ubuntu 16.04

    Pytorch

    架構版本

    CUDA版本（僅GPU機型）

    作業系統

    • Pytorch2.1

    • Pytorch2.0

    • Pytorch1.8

    • Pytorch1.7

    • Pytorch1.12

    • Pytorch1.11

    • Pytorch1.10

    • CUDA 121

    • CUDA 117

    • CUDA 114

    • CUDA 113

    • CUDA 101

    • Ubuntu 22.04

    • Ubuntu 20.04

    • Ubuntu 18.04

    DeepRec

    架構版本

    CUDA版本（僅GPU機型）

    作業系統

    • DeepRec2302

    • DeepRec2212

    CUDA 114

    Ubuntu 18.04

    XGBoost

    架構版本

    CUDA版本（僅GPU機型）

    作業系統

    XGBoost1.6.0

    不涉及，僅支援CPU機型

    Ubuntu 18.04

    TritonServer

    架構版本

    CUDA版本（僅GPU機型）

    作業系統

    • TritonServer23.02

    • TritonServer21.09

    • CUDA 120

    • CUDA 114

    Ubuntu 20.04

    核心鏡像列表

    靈駿智算資源（Serverless版）適用的鏡像

    鏡像名稱

    架構

    機型

    CUDA

    作業系統

    地區

    開發語言&版本

    deepspeed-training:23.06-gpu-py310-cu121-ubuntu22.04

    • PyTorch 2.1

    • Megatron-LM 23.06

    • DeepSpeed 0.9.5

    • Transformers 4.29.2

    • Nemo 1.19.0

    GPU

    121

    ubuntu22.04

    華北6（烏蘭察布）

    Python3.10

    megatron-training:23.06-gpu-py310-cu121-ubuntu22.04

    • PyTorch 2.1

    • Megatron-LM 23.06

    • DeepSpeed 0.9.5

    • Transformers 4.29.2

    • Nemo 1.19.0

    GPU

    121

    ubuntu22.04

    華北6（烏蘭察布）

    Python3.10

    nemo-training:23.06-gpu-py310-cu121-ubuntu22.04

    • PyTorch 2.1

    • Megatron-LM 23.06

    • DeepSpeed 0.9.5

    • Transformers 4.29.2

    • Nemo 1.19.0

    GPU

    121

    ubuntu22.04

    華北6（烏蘭察布）

    Python3.10

    AIGC相關鏡像

    鏡像名稱

    架構

    機型

    CUDA

    作業系統

    支援地區

    開發語言&版本

    stable-diffusion-webui:3.0

    StableDiffusionWebUI3.0

    GPU

    117

    ubuntu22.04

    • 華東1（杭州）

    • 華東2（上海）

    • 華北2（北京）

    • 華北3（張家口）

    • 華北6（烏蘭察布）

    • 華南1（深圳）

    • 華南2（河源）

    • 西南1（成都）

    Python3.10

    stable-diffusion-webui:2.2

    StableDiffusionWebUI2.2

    GPU

    117

    ubuntu22.04

    Python3.10

    stable-diffusion-webui:1.1

    StableDiffusionWebUI1.1

    GPU

    117

    ubuntu22.04

    Python3.10

    stable-diffusion-webui-env:pytorch1.13-gpu-py310-cu117-ubuntu22.04

    SD-WebUI-ENV

    GPU

    117

    ubuntu22.04

    Python3.10

    EAS主推的鏡像

    阿里雲PAI提供的EAS產品部分官方鏡像列表如下所示。您可以前往PAI控制台的鏡像AI資產頁面查看更完整的鏡像列表資訊。以下列出的鏡像地址以華東1（杭州）地區為例。

    鏡像名稱

    架構

    鏡像描述

    鏡像地址

    chat-llm-webui:3.0-blade

    • Blade 0.4.3.dev118+gea4a4a1

    • PyTorch 2.1.0+cu118

    • Transformers 4.36.2

    用於EAS推理鏡像，使用Blade後端為大型語言模型（LLM）提供推理服務，該推理服務同時支援WebUI與API訪問。

    eas-registry-vpc.cn-hangzhou.cr.aliyuncs.com/pai-eas/chat-llm-webui:3.0-blade

    chatbot-langchain:1.0

    ChatbotLangChain 1.0

    基於LangChain搭建的知識庫問答，用於EAS推理鏡像。

    eas-registry-vpc.cn-hangzhou.cr.aliyuncs.com/pai-eas/chatbot-langchain:1.0

    comfyui:0.2-api

    ComfyUI 0.2

    基於ComfyUI的EAS鏡像，用於文生圖、圖生圖等非同步API調用。

    eas-registry-vpc.cn-hangzhou.cr.aliyuncs.com/pai-eas/comfyui:0.2

    comfyui:0.2

    ComfyUI 0.2

    基於ComfyUI的EAS鏡像，用於文生圖、圖生圖等情境。

    eas-registry-vpc.cn-hangzhou.cr.aliyuncs.com/pai-eas/comfyui:0.2

    comfyui:0.2-cluster

    ComfyUI 0.2

    基於ComfyUI的EAS鏡像，用於文生圖、圖生圖等情境。

    eas-registry-vpc.cn-hangzhou.cr.aliyuncs.com/pai-eas/comfyui:0.2

    kohya_ss:2.2

    Kohya 2.2

    基於Kohya的EAS鏡像，用於部署Stable Diffusion模型微調應用。

    eas-registry-vpc.cn-hangzhou.cr.aliyuncs.com/pai-eas/kohya_ss:2.2

    modelscope-inference:1.9.1

    ModelScope 1.9.1

    用於部署Modelscope模型。

    eas-registry-vpc.cn-hangzhou.cr.aliyuncs.com/pai-eas/modelscope-inference:1.9.1

    stable-diffusion-webui:4.2-cluster-webui

    StableDiffusionWebUI 4.2

    基於Stable Diffusion WebUI的EAS鏡像，用於文生圖、圖生圖等情境，並且支援多使用者同時訪問和使用者間資源隔離。

    eas-registry-vpc.cn-hangzhou.cr.aliyuncs.com/pai-eas/stable-diffusion-webui:4.2

    stable-diffusion-webui:4.2-api

    StableDiffusionWebUI 4.2

    基於Stable Diffusion WebUI的EAS鏡像，用於文生圖、圖生圖等非同步API調用。

    eas-registry-vpc.cn-hangzhou.cr.aliyuncs.com/pai-eas/stable-diffusion-webui:4.2

    stable-diffusion-webui:4.2-standard

    StableDiffusionWebUI 4.2

    基於Stable Diffusion WebUI的EAS鏡像，用於文生圖、圖生圖等情境。

    eas-registry-vpc.cn-hangzhou.cr.aliyuncs.com/pai-eas/stable-diffusion-webui:4.2

    tensorflow-serving:2.14.1

    TensorflowServing 2.14.1

    基於開源TensorFlow Serving鏡像的EAS鏡像，用於將TensorFlow模型部署為推理服務，僅適用於CPU環境。

    eas-registry-vpc.cn-hangzhou.cr.aliyuncs.com/pai-eas/tensorflow_serving:2.14.1

    tensorflow-serving:2.14.1-gpu

    TensorflowServing 2.14.1

    基於開源TensorFlow Serving鏡像的EAS鏡像，用於將TensorFlow模型部署為推理服務，適用於GPU環境。

    eas-registry-vpc.cn-hangzhou.cr.aliyuncs.com/pai-eas/tensorflow_serving:2.14.1-gpu

    chat-llm-webui:3.0

    • PyTorch 2.0.1

    • Transformers 4.33.3

    用於EAS推理鏡像，使用HuggingFace後端為大型語言模型（LLM）提供推理服務，該推理服務同時支援WebUI與API訪問。

    eas-registry-vpc.cn-hangzhou.cr.aliyuncs.com/pai-eas/chat-llm-webui:3.0

    chat-llm-webui:3.0-vllm

    • PyTorch 2.1.2

    • Transformers 4.36.2

    • VLLM 0.2.7

    用於EAS推理鏡像，使用VLLM後端為大型語言模型（LLM）提供推理服務，該推理服務同時支援WebUI與API訪問。

    eas-registry-vpc.cn-hangzhou.cr.aliyuncs.com/pai-eas/chat-llm-webui:3.0-vllm

    huggingface-inference:1.0-transformers4.33

    Transformers 4.33

    用於部署HuggingFace模型。

    eas-registry-vpc.cn-hangzhou.cr.aliyuncs.com/pai-eas/huggingface-inference:transformers-4.33

    tritonserver:23.11-py3

    TritonServer 23.11

    基於開源TritonServer鏡像的EAS鏡像，用於將模型部署為推理服務。

    eas-registry-vpc.cn-hangzhou.cr.aliyuncs.com/pai-eas/tritonserver:23.11-py3