6295d2e9bc
### What this PR does / why we need it? This pr upgrades torch-npu to 0320, so that #321, https://github.com/vllm-project/vllm-ascend/issues/267#issuecomment-2745045743 could be fixed, and #372 should be reverted after this pr ### Does this PR introduce _any_ user-facing change? upgrade torch-npu to 0320 ### How was this patch tested? tested locally with long seq inferencing. --------- Signed-off-by: MengqingCao <cmq0113@163.com>
16 lines
530 B
Bash
Executable File
16 lines
530 B
Bash
Executable File
#!/bin/bash
|
|
set -ex
|
|
mkdir pta
|
|
cd pta || exit
|
|
wget https://pytorch-package.obs.cn-north-4.myhuaweicloud.com/pta/Daily/v2.5.1/20250320.3/pytorch_v2.5.1_py310.tar.gz
|
|
tar -zxvf pytorch_v2.5.1_py310.tar.gz
|
|
|
|
if [ "$(uname -i)" == "aarch64" ]
|
|
then
|
|
pip install ./torch_npu-2.5.1.dev20250320-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
|
|
else
|
|
pip install ./torch_npu-2.5.1.dev20250320-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl --extra-index https://download.pytorch.org/whl/cpu/
|
|
fi
|
|
|
|
cd ..
|
|
rm -rf pta |