安装

查看源文件

版本匹配关系

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

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

MindFormers

MindSpore

CANN

固件与驱动

镜像链接

1.3.0

2.4.0

8.0.RC3.beta1

24.1.RC3

Link

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

历史版本配套关系:

MindFormers

MindSpore

CANN

固件与驱动

镜像链接

1.2.0

2.3.0

8.0.RC2.beta1

24.1.RC2

Link

环境安装

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

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

MindFormers安装

MindFormers支持源码编译安装和pip安装两种方式。

源码编译安装

用户可以执行如下命令编译并安装MindFormers:

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

pip安装

用户可以执行如下命令下载并安装MindFormers:

pip install mindformers==1.3.0

注意:此方式只安装了MindFormers基础软件包,模型文件和脚本等请从MindFormers gitee仓库中获取。

安装验证

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

import mindformers as mf
mf.run_check()

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

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