[0.11.0]Chery pick pta upgrade change (#3940)
This PR cherry-pick two commit from main to upgrade torch-npu to 2.7.1 official release --------- Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>
This commit is contained in:
@@ -10,7 +10,7 @@ pyyaml
|
||||
scipy
|
||||
setuptools>=64
|
||||
setuptools-scm>=8
|
||||
torch>=2.7.1
|
||||
torch==2.7.1
|
||||
torchvision
|
||||
wheel
|
||||
opencv-python-headless<=4.11.0.86 # Required to avoid numpy version conflict with vllm
|
||||
@@ -23,6 +23,6 @@ quart
|
||||
numba
|
||||
|
||||
# Install torch_npu
|
||||
--pre
|
||||
--extra-index-url https://mirrors.huaweicloud.com/ascend/repos/pypi
|
||||
torch-npu==2.7.1.dev20250724
|
||||
#--pre
|
||||
#--extra-index-url https://mirrors.huaweicloud.com/ascend/repos/pypi
|
||||
torch-npu==2.7.1
|
||||
|
||||
Reference in New Issue
Block a user