Skip to content

Latest commit

 

History

History
38 lines (29 loc) · 1.88 KB

environment.zh.md

File metadata and controls

38 lines (29 loc) · 1.88 KB

昇腾NPU环境准备

依赖

需求 支持的版本 推荐版本 注意
vLLM main main vllm-ascend 依赖
Python >= 3.9 3.10 vllm 依赖
CANN >= 8.0.RC2 8.0.RC3 vllm-ascend and torch-npu 依赖
torch-npu >= 2.4.0 2.5.1rc1 vllm-ascend 依赖
torch >= 2.4.0 2.5.1 torch-npu and vllm 依赖

以下为安装推荐版本软件的简短说明:

容器化安装

您可以直接使用容器镜像,只需一行命令即可:

docker run \
    --name vllm-ascend-env \
    --device /dev/davinci1 \
    --device /dev/davinci_manager \
    --device /dev/devmm_svm \
    --device /dev/hisi_hdc \
    -v /usr/local/dcmi:/usr/local/dcmi \
    -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
    -v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ \
    -v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info \
    -v /etc/ascend_install.info:/etc/ascend_install.info \
    -it quay.io/ascend/cann:8.0.rc3.beta1-910b-ubuntu22.04-py3.10 bash

您无需手动安装 torchtorch_npu ,它们将作为 vllm-ascend 依赖项自动安装。

手动安装

您也可以选择手动安装,按照昇腾安装指南中提供的说明配置环境。