From e14f5b37e4cf6086d80e600c88a4e2e402a74a72 Mon Sep 17 00:00:00 2001 From: MengqingCao Date: Wed, 29 May 2024 09:53:31 +0000 Subject: [PATCH] update cann kernels url --- README.md | 2 +- README_zh.md | 2 +- setup.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index ca70110f..14a8f4a2 100644 --- a/README.md +++ b/README.md @@ -359,7 +359,7 @@ chmod +x Ascend-cann-toolkit_8.0.RC1.alpha001_linux-"$(uname -i)".run ./Ascend-cann-toolkit_8.0.RC1.alpha001_linux-"$(uname -i)".run --install # install CANN Kernels -wget https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL%20V100R001C18B800TP015/Ascend-cann-kernels-910b_8.0.RC1.alpha001_linux.run +wget https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL%20V100R001C17SPC701/Ascend-cann-kernels-910b_8.0.RC1.alpha001_linux.run chmod +x Ascend-cann-kernels-910b_8.0.RC1.alpha001_linux.run ./Ascend-cann-kernels-910b_8.0.RC1.alpha001_linux.run --install diff --git a/README_zh.md b/README_zh.md index 4f659588..66a4c031 100644 --- a/README_zh.md +++ b/README_zh.md @@ -359,7 +359,7 @@ chmod +x Ascend-cann-toolkit_8.0.RC1.alpha001_linux-"$(uname -i)".run ./Ascend-cann-toolkit_8.0.RC1.alpha001_linux-"$(uname -i)".run --install # 安装 CANN Kernels -wget https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL%20V100R001C18B800TP015/Ascend-cann-kernels-910b_8.0.RC1.alpha001_linux.run +wget https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL%20V100R001C17SPC701/Ascend-cann-kernels-910b_8.0.RC1.alpha001_linux.run chmod +x Ascend-cann-kernels-910b_8.0.RC1.alpha001_linux.run ./Ascend-cann-kernels-910b_8.0.RC1.alpha001_linux.run --install diff --git a/setup.py b/setup.py index 87620c37..f72eefb6 100644 --- a/setup.py +++ b/setup.py @@ -33,7 +33,7 @@ extra_require = { "qwen": ["tiktoken", "transformers_stream_generator"], "modelscope": ["modelscope"], "quality": ["ruff"], - "torch_npu": ["torch==2.2.0", "torch_npu==2.2.0", "decorator"] + "torch_npu": ["torch==2.1.0", "torch_npu==2.1.0", "decorator"] }