inference-nv-pytorch 26.01

更新时间:
复制为 MD 格式

本文介绍inference-nv-pytorch 26.01版本发布记录。

Main Features and Bug Fix Lists

Main Features

  • 提供了CUDA12.8CUDA13.0两个CUDA版本的镜像,其中:

    • CUDA12.8镜像仅支持amd64架构。

    • CUDA13.0镜像支持amd64aarch64架构。

  • CUDA12.8镜像deepgpu-comfyui 升级至1.4.1,deepgpu-torch优化组件升级至0.1.18+torch2.9.0cu128。

  • CUDA12.8CUDA13.0镜像vLLM版本升级到v0.14.0,SGLang版本升级至v0.5.7

Bug Fix

Contents

镜像名称

inference-nv-pytorch

Tag

26.01-vllm0.14.0-pytorch2.9-cu128-20260121-serverless

26.01-sglang0.5.7-pytorch2.9-cu128-20260113-serverless

26.01-vllm0.14.0-pytorch2.9-cu130-20260123-serverless

26.01-sglang0.5.7-pytorch2.9-cu130-20260113-serverless

支持架构

amd64

amd64

amd64

aarch64

amd64

aarch64

应用场景

大模型推理

大模型推理

大模型推理

大模型推理

大模型推理

大模型推理

框架

pytorch

pytorch

pytorch

pytorch

pytorch

pytorch

Requirements

NVIDIA Driver release >= 570

NVIDIA Driver release >= 570

NVIDIA Driver release >= 580

NVIDIA Driver release >= 580

NVIDIA Driver release >= 580

NVIDIA Driver release >= 580

系统组件

  • Ubuntu 24.04

  • Python 3.12

  • Torch 2.9.1

  • CUDA 12.8

  • diffusers 0.36.0

  • deepgpu-comfyui 1.4.1

  • deepgpu-torch 0.1.18+torch2.9.0cu128

  • flash_attn 2.8.3

  • flashinfer-python 0.5.3

  • imageio 2.37.2

  • imageio-ffmpeg 0.6.0

  • ray 2.53.0

  • transformers 4.57.6

  • triton 3.5.1

  • torchaudio 2.9.1

  • torchvision 0.24.1

  • vllm 0.14.0

  • xfuser 0.4.5

  • xgrammar 0.1.27

  • ljperf 0.1.0+477686c5

  • Ubuntu 24.04

  • Python 3.12

  • Torch 2.9.1+cu128

  • CUDA 12.8

  • torchaudio 2.9.1+128

  • torchvision 0.24.1+128

  • diffusers 0.36.0

  • decord 0.6.0

  • decord2 3.0.0

  • deepgpu-comfyui 1.4.1

  • deepgpu-torch 0.1.18+torch2.9.0cu128

  • flash_attn 2.8.3

  • flash_mla 1.0.0+1408756

  • flashinfer-python 0.5.3

  • imageio 2.37.2

  • imageio-ffmpeg 0.6.0

  • ray 2.53.0

  • transformers 4.57.1

  • sgl-kernel 0.3.20

  • sglang 0.5.7

  • xgrammar 0.1.27

  • triton 3.5.1

  • torchao 0.9.0

  • xfuser 0.4.5

  • ljperf 0.1.0+477686c5

  • Ubuntu 24.04

  • Python 3.12

  • Torch 2.9.1+cu130

  • CUDA 13.0.2

  • diffusers 0.36.0

  • flash_attn 2.8.3

  • flashinfer-python 0.5.3

  • imageio 2.37.2

  • imageio-ffmpeg 0.6.0

  • ray 2.53.1

  • transformers 4.57.6

  • triton 3.5.0

  • torchaudio 2.9.1+cu130

  • torchvision 0.24.1+cu130

  • vllm 0.14.0

  • xfuser 0.4.5

  • xgrammar 0.1.27

  • ljperf 0.1.0+d0e4a408

  • Ubuntu 24.04

  • Python 3.12

  • Torch 2.9.1+cu130

  • CUDA 13.0.2

  • diffusers 0.36.0

  • flash_attn 2.8.3

  • flashinfer-python 0.5.3

  • transformers 4.57.6

  • ray 2.53.0

  • vllm 0.14.0

  • triton 3.5.1

  • torchaudio 2.9.1+cu130

  • torchvision 2.9.1+cu130

  • xfuser 0.4.5

  • xgrammar 0.1.27

  • ljperf 0.1.0+477686c5

  • Ubuntu 24.04

  • Python 3.12

  • Torch 2.9.1+cu130

  • CUDA 13.0.2

  • diffusers 0.36.0

  • decord 0.6.0

  • decord2 3.0.0

  • flash_attn 2.8.3

  • flashinfer-python 0.5.3

  • imageio 2.37.2

  • imageio-ffmpeg 0.6.0

  • ray 2.53.0

  • transformers 4.57.1

  • sgl-kernel 0.3.20

  • sglang 0.5.7

  • xgrammar 0.1.27

  • triton 3.5.1

  • torchao 0.9.0

  • torchaudio 2.9.1

  • torchvision 0.24.1+cu130

  • xfuser 0.4.5

  • ljperf 0.1.0+d0e4a408

  • Ubuntu 24.04

  • Python 3.12

  • Torch 2.9.1+cu130

  • CUDA 13.0.2

  • diffusers 0.36.0

  • decord2 3.0.0

  • flash_attn 2.8.3

  • flashinfer-python 0.5.3

  • imageio 2.37.2

  • imageio-ffmpeg 0.6.0

  • transformers 4.57.1

  • sgl-kernel 0.3.20

  • sglang 0.5.7

  • xgrammar 0.1.27

  • triton 3.5.1

  • torchao 0.9.0

  • torchaudio 2.9.1

  • torchvision 0.24.1

  • xfuser 0.4.5

Asset

公网镜像

CUDA12.8 Asset

  • egslingjun-registry.cn-wulanchabu.cr.aliyuncs.com/egslingjun/inference-nv-pytorch:26.01-vllm0.14.0-pytorch2.9-cu128-20260121-serverless

  • egslingjun-registry.cn-wulanchabu.cr.aliyuncs.com/egslingjun/inference-nv-pytorch:26.01-sglang0.5.7-pytorch2.9-cu128-20260113-serverless

CUDA13.0 Asset

  • egslingjun-registry.cn-wulanchabu.cr.aliyuncs.com/egslingjun/inference-nv-pytorch:26.01-vllm0.14.0-pytorch2.9-cu130-20260123-serverless

  • egslingjun-registry.cn-wulanchabu.cr.aliyuncs.com/egslingjun/inference-nv-pytorch:26.01-sglang0.5.7-pytorch2.9-cu130-20260113-serverless

VPC镜像

将指定的AI容器镜像Asset URIegslingjun-registry.cn-wulanchabu.cr.aliyuncs.com/egslingjun/{image:tag}替换为acs-registry-vpc.{region-id}.cr.aliyuncs.com/egslingjun/{image:tag}即可在VPC内快速拉取ACS AI容器镜像。

  • {region-id}:ACS产品开服地域的地域ID。例如:cn-beijingcn-wulanchabu等。

  • {image:tag}:AI容器镜像的名称和Tag。例如:inference-nv-pytorch:25.10-vllm0.11.0-pytorch2.8-cu128-20251028-serverlesstraining-nv-pytorch:25.10-serverless等。

说明

镜像适用于ACS产品形态、灵骏多租产品形态,不适用于灵骏单租产品形态,请勿在灵骏单租场景使用。

Driver Requirements

  • CUDA12.8:NVIDIA Driver release >= 570

  • CUDA13.0:NVIDIA Driver release >= 580

Quick Start

以下示例内容仅通过Docker方式拉取inference-nv-pytorch镜像,并使用Qwen2.5-7B-Instruct模型测试推理服务。

说明

ACS中使用inference-nv-pytorch镜像需要通过控制台创建工作负载界面的制品中心页面选取,或者通过YAML文件指定镜像引用。更多详细操作,请参见使用ACS GPU算力构建模型推理服务系列内容:

  1. 拉取推理容器镜像。

    docker pull egslingjun-registry.cn-wulanchabu.cr.aliyuncs.com/egslingjun/inference-nv-pytorch:[tag]
  2. 通过ModelScope下载开源模型。

    pip install modelscope
    cd /mnt
    modelscope download --model Qwen/Qwen2.5-7B-Instruct --local_dir ./Qwen2.5-7B-Instruct
  3. 启动以下命令进入容器。

    docker run -it --rm --gpus all --network=host --privileged --init --ipc=host \
    --ulimit memlock=-1 --ulimit stack=67108864  \
    -v /mnt/:/mnt/ \
    egslingjun-registry.cn-wulanchabu.cr.aliyuncs.com/egslingjun/inference-nv-pytorch:[tag]
  4. 执行推理测试,测试vLLM推理对话功能。

    1. 启动Server端服务。

      python3 -m vllm.entrypoints.openai.api_server \
      --model /mnt/Qwen2.5-7B-Instruct \
      --trust-remote-code --disable-custom-all-reduce \
      --tensor-parallel-size 1
    2. Client端进行测试。

      curl http://localhost:8000/v1/chat/completions \
          -H "Content-Type: application/json" \
          -d '{
          "model": "/mnt/Qwen2.5-7B-Instruct",  
          "messages": [
          {"role": "system", "content": "你是个友善的AI助手。"},
          {"role": "user", "content": "介绍一下深度学习。"}
          ]}'

      输出:

      {"id":"chat-d3c28759793d4376a65bfc4e40b59a71","object":"chat.completion","created":1735278194,"model":"/mnt/deep_learning_test/testsuite/dataset/llms_inference_qwen7b-v2.5_accelerate/checkpoint/7B-V2.5/","choices":[{"index":0,"message":{"role":"assistant","content":"深度学习是机器学习的一个分支,它受到生物神经系统,尤其是大脑中神经元之间交互的启发。深度学习利用深层神经网络处理并分析大量数据,以识别出有效的预测模型。这种技术在图像识别、语言处理、自然语言处理等众多领域上取得了显著的成功。\n\n在深度学习中,神经网络由多层组成,包括输入层、若干隐藏层和输出层。每一层都包含多个节点(或称为神经元),这些节点通过加权连接与其他层的节点相连接。在训练过程中,神经网络会根据输入数据调整这些连接的权重,以最小化预测输出与实际标签之间的误差。这一过程通常使用梯度下降等优化算法实现。\n\n深度学习模型的训练需要大量的计算资源和数据支持。近年来,随着计算硬件的进步(如GPU、TPU等),以及数据集的快速增长,深度学习技术得到了广泛的应用和发展。除了上述提到的应用外,深度学习还被广泛应用于医疗诊断、自动驾驶、游戏和金融等多个领域。","tool_calls":[]},"logprobs":null,"finish_reason":"stop","stop_reason":null}],"usage":{"prompt_tokens":237,"completion_tokens":213}}

      更多关于vLLM的使用方法请参见vLLM

Known Issues

  • deepgpu-comfyui插件,加速Wanx模型视频生成,目前仅支持GN8IS、G49E、G59卡型。