下载
中文
注册

准备软件包

如果使用容器化安装MindIE,则跳过安装开发环境步骤,请直接前往容器化部署和镜像制作章节通过镜像安装或者容器化安装MindIE。

软件包配套关系

MindIE与CANN版本必须配套使用,其版本配套关系如表1所示。

表1 版本配套列表

MindIE版本

配套CANN版本

1.0.RC3

8.0.RC3

下载软件包

软件安装前,请参考表2获取所需软件包和对应的数字签名文件,各软件包版本号需要保持一致。

下载本软件即表示您同意华为企业业务最终用户许可协议(EULA)的条款和条件。

表2 MindIE依赖软件

依赖

软件包

软件说明

获取链接

CANN

Ascend-cann-toolkit_{version}_linux-{arch}.run

开发套件包。

获取链接

Ascend-cann-kernels-{chip_type}_{version}_linux-{arch}.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 配置与安装》中的“(可选)安装APEX模块”章节,根据Python3.10版本自行编译。

torch-{torch_version}+cpu-cp310-cp310-linux_{arch}.whl

PyTorch框架whl包。

说明:

当前只支持1.11.0和2.1.0版本。

请从《Ascend Extension for PyTorch 配置与安装》中的“安装PyTorch框架”章节获取。

由于Pytorch1.11.0官方版本存在安全漏洞,因此强烈不建议使用Pytorch1.11.0官方版本,若必须使用建议先修复相关漏洞,避免引入安全风险。