下载软件包
软件安装前，请参考表2获取所需软件包和对应的数字签名文件，各软件包版本号需要保持一致。
下载本软件即表示您同意华为企业业务最终用户许可协议（EULA）的条款和条件。
依赖 |
软件包 |
软件说明 |
获取链接 |
|---|---|---|---|
CANN |
Ascend-cann-toolkit_{version}_linux-{arch}.run |
开发套件包。 |
|
Ascend-cann-kernels-{chip_type}_{version}_linux.run |
二进制算子包。 |
||
Ascend-cann-nnal_{version}_linux-{arch}.run |
加速库软件包。 |
||
ATB Models |
Ascend-mindie-atb-models_{version}_linux-{arch}-{abi}.tar.gz |
模型库安装包。 |
|
Ascend Extension for PyTorch |
torch_npu-{torch_version}.post{post_id}-cp310-cp310-linux_{arch}.whl |
torch_npu插件whl包。 |
说明：
在Pytorch栏单击对应版本后方下载按钮，跳转至Pytorch的gitee仓发布页，然后在页面下方获取对应版本的torch_npu。 |
apex-{apex_version}_ascend-cp310-cp310-{arch}.whl |
Apex模块的whl包。 |
请参见《Ascend Extension for PyTorch 配置与安装》中的章节根据Python3.10版本自行编译。 |
|
torch-{torch_version}+cpu-cp310-cp310-linux_{arch}.whl |
PyTorch框架whl包。 说明：
当前只支持1.11.0和2.1.0版本。 |
请从《Ascend Extension for PyTorch 配置与安装》中的获取。 |
由于Pytorch1.11官方版本存在安全漏洞，因此强烈不建议使用Pytorch1.11官方版本，若必须使用建议先修复相关漏洞，避免引入安全风险。