[NPU] fix CI (#26390)

This commit is contained in:
Makcum888e
2026-05-27 09:45:42 +03:00
committed by GitHub
parent f32ca1e0e5
commit 9060509214
2 changed files with 2 additions and 2 deletions
+2 -1
View File
@@ -57,7 +57,8 @@ if [ -n "$OPTIONAL_DEPS" ]; then
else
PYTORCH_VERSION="2.8.0"
TORCHVISION_VERSION="0.23.0"
${UV_PIP_INSTALL} torch==${PYTORCH_VERSION} torchvision==${TORCHVISION_VERSION} --index-url ${TORCH_CACHE_URL:="https://download.pytorch.org/whl/cpu"} --extra-index-url ${PYPI_CACHE_URL:="https://pypi.org/simple/"}
TORCHAUDIO_VERSION="2.8.0"
${UV_PIP_INSTALL} torch==${PYTORCH_VERSION} torchvision==${TORCHVISION_VERSION} torchaudio==${TORCHAUDIO_VERSION} --index-url ${TORCH_CACHE_URL:="https://download.pytorch.org/whl/cpu"} --extra-index-url ${PYPI_CACHE_URL:="https://pypi.org/simple/"}
PTA_URL="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"
${PIP_INSTALL} ${PTA_URL}
fi