minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[NPU] Add torchaudio dependency for NPU platform (#26267)

Browse Source
This commit is contained in:
xdtbynd
2026-05-25 16:30:12 +08:00
committed by GitHub
parent 81704ad602
commit 0942011665
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/npu.Dockerfile
+2 -1
View File
@@ -13,6 +13,7 @@ ARG PIP_INDEX_URL="https://pypi.org/simple/"
ARG APTMIRROR="" ARG APTMIRROR=""
ARG PYTORCH_VERSION="2.8.0" ARG PYTORCH_VERSION="2.8.0"
ARG TORCHVISION_VERSION="0.23.0" ARG TORCHVISION_VERSION="0.23.0"
ARG TORCHAUDIO_VERSION="2.8.0"
ARG PTA_URL_ARM64="https://gitcode.com/Ascend/pytorch/releases/download/v7.3.0-pytorch2.8.0/torch_npu-2.8.0.post2-cp311-cp311-manylinux_2_28_aarch64.whl" ARG PTA_URL_ARM64="https://gitcode.com/Ascend/pytorch/releases/download/v7.3.0-pytorch2.8.0/torch_npu-2.8.0.post2-cp311-cp311-manylinux_2_28_aarch64.whl"
ARG PTA_URL_AMD64="https://gitcode.com/Ascend/pytorch/releases/download/v7.3.0-pytorch2.8.0/torch_npu-2.8.0.post2-cp311-cp311-manylinux_2_28_x86_64.whl" ARG PTA_URL_AMD64="https://gitcode.com/Ascend/pytorch/releases/download/v7.3.0-pytorch2.8.0/torch_npu-2.8.0.post2-cp311-cp311-manylinux_2_28_x86_64.whl"
ARG SGLANG_TAG=main ARG SGLANG_TAG=main
@@ -78,7 +79,7 @@ RUN ${PIP_INSTALL} sglang-router
### Install PyTorch and PTA ### Install PyTorch and PTA
RUN . /etc/environment_new && \ RUN . /etc/environment_new && \
(${PIP_INSTALL} torch==${PYTORCH_VERSION} torchvision==${TORCHVISION_VERSION} --index-url https://download.pytorch.org/whl/cpu) \ (${PIP_INSTALL} torch==${PYTORCH_VERSION} torchvision==${TORCHVISION_VERSION} torchaudio==${TORCHAUDIO_VERSION} --index-url https://download.pytorch.org/whl/cpu) \
&& (${PIP_INSTALL} ${PTA_URL}) && (${PIP_INSTALL} ${PTA_URL})
python/pyproject_npu.toml
+1
View File
@@ -57,6 +57,7 @@ dependencies = [
"tiktoken", "tiktoken",
"timm==1.0.16", "timm==1.0.16",
"torchao==0.9.0", "torchao==0.9.0",
"torchaudio==2.8.0",
"tqdm", "tqdm",
"mistral_common>=1.11.0", "mistral_common>=1.11.0",
"transformers==5.8.1", "transformers==5.8.1",