安装

查看源文件

版本匹配关系

当前支持的硬件为Atlas 800T A2训练服务器。

当前套件建议使用的Python版本为3.10。

MindFormers

MindSpore

CANN

固件与驱动

镜像链接

在研版本

在研版本

在研版本

在研版本

不涉及

当前MindFormers建议使用如上的软件配套关系。

历史版本配套关系:

MindFormers

MindSpore

CANN

固件与驱动

镜像链接

r1.3.0

2.4.0

8.0.RC3.beta1

24.1.RC3

Link

r1.2.0

2.3.0

8.0.RC2.beta1

24.1.RC2

Link

环境安装

  1. 安装固件与驱动:通过版本匹配关系中的固件与驱动链接下载安装包,参考昇腾官方教程进行安装。

  2. 安装CANN和MindSpore:使用官方提供的Docker镜像(镜像中已包含CANN、MindSpore,无需手动安装)或者按照MindSpore官网的手动安装章节进行安装。

MindFormers安装

目前仅支持源码编译安装,用户可以执行如下命令安装MindFormers:

git clone -b dev https://gitee.com/mindspore/mindformers.git
cd mindformers
bash build.sh

安装验证

判断MindFormers是否安装成功可以执行以下命令:

import mindformers as mf
mf.run_check()

出现以下类似结果,证明安装成功:

- INFO - All checks passed, used **** seconds, the environment is correctly set up!