inference-nv-pytorch 25.06

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

Main Features and Bug Fix Lists

Main Features

  • vLLM版本升级到v0.9.0.1

  • SGLang版本升级至v0.4.7

  • 新增deepgpu-comfyui插件,支持在L20上加速ComfyUI服务进行Wan2.1FLUX模型推理,性能相比于torch提升8%~40%。

Bug Fix

(暂无)

Contents

inference-nv-pytorch

inference-nv-pytorch

镜像Tag

25.06-vllm0.9.0.1-pytorch2.7-cu128-20250609-serverless

25.06-sglang0.4.7-pytorch2.7-cu128-20250611-serverless

应用场景

大模型推理

大模型推理

框架

pytorch

pytorch

Requirements

NVIDIA Driver release >= 570

NVIDIA Driver release >= 550

系统组件

  • Ubuntu 24.04

  • Python 3.12

  • Torch 2.7.1+cu128

  • CUDA 12.8

  • NCCL 2.27.3

  • accelerate 1.7.0

  • diffusers 0.33.1

  • deepgpu-comfyui 1.1.5

  • deepgpu-torch 0.0.21+torch2.7.0cu128

  • flash_attn 2.7.4.post1

  • imageio 2.37.0

  • imageio-ffmpeg 0.6.0

  • ray 2.46.0

  • transformers 4.52.4

  • vllm 0.9.0.2.dev0+g5fbbfe9a4.d20250609

  • xgrammar 0.1.19

  • triton 3.3.1

  • Ubuntu 22.04

  • Python 3.10

  • Torch 2.7.1+cu128

  • CUDA 12.8

  • NCCL 2.27.3

  • accelerate 1.7.0

  • diffusers 0.33.1

  • deepgpu-comfyui 1.1.5

  • deepgpu-torch 0.0.21+torch2.7.0cu128

  • flash_attn 2.7.4.post1

  • flash_mla 1.0.0+9edee0c

  • flashinfer-python 0.2.6.post1

  • imageio 2.37.0

  • imageio-ffmpeg 0.6.0

  • ray 2.46.0

  • transformers 4.52.3

  • sgl-kernel 0.1.7

  • sglang 0.4.7

  • xgrammar 0.1.19

  • triton 3.3.1

  • torchao 0.9.0

Asset

公网镜像

  • egslingjun-registry.cn-wulanchabu.cr.aliyuncs.com/egslingjun/inference-nv-pytorch:25.06-vllm0.9.0.1-pytorch2.7-cu128-20250609-serverless

  • egslingjun-registry.cn-wulanchabu.cr.aliyuncs.com/egslingjun/inference-nv-pytorch:25.06-sglang0.4.7-pytorch2.7-cu128-20250611-serverless

VPC镜像

  • acs-registry-vpc.{region-id}.cr.aliyuncs.com/egslingjun/{image:tag}

    {region-id}为您使用的ACS产品所在的开服地域,比如:cn-beijing、cn-wulanchabu等。
    {image:tag}为实际镜像的名称和Tag。
说明

25.06-vllm0.9.0.1-pytorch2.7-cu128-20250609-serverless25.06-sglang0.4.7-pytorch2.7-cu128-20250611-serverless镜像适用于ACS产品形态、灵骏多租产品形态,不适用于灵骏单租产品形态。

Driver Requirements

cuda 12.8的镜像:NVIDIA Driver release >= 570

Quick Start

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

说明

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

  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 -d -t --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. 启动Sever服务。

      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": "介绍一下深度学习。"}
          ]}'

      输出如下所示:

      image.png

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

Known Issues

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