安装指南

查看源文件

确认版本匹配关系

当前支持的硬件为 Atlas 800T A2、Atlas 800I A2、Atlas 900 A3 SuperPoD。

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

MindSpore Transformers

MindSpore

CANN

固件与驱动

1.6.0

2.7.0

8.2.RC1

25.2.0

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

历史版本配套关系:

MindSpore Transformers

MindSpore

CANN

固件与驱动

1.5.0

2.6.0-rc1

8.1.RC1

25.0.RC1

1.3.2

2.4.10

8.0.0

24.1.0

1.3.0

2.4.0

8.0.RC3

24.1.RC3

1.2.0

2.3.0

8.0.RC2

24.1.RC2

安装依赖软件

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

  2. 安装CANN和MindSpore:按照MindSpore官网的手动安装章节进行安装。

安装MindSpore Transformers

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

源码编译方式安装

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

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

pip方式安装

用户可以执行如下命令安装MindSpore Transformers:

pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/2.7.0rc1/MindFormers/any/mindformers-1.6.0-py3-none-any.whl --trusted-host ms-release.obs.cn-north-4.myhuaweicloud.com -i https://repo.huaweicloud.com/repository/pypi/simple
  • 此安装方式需要访问公网,如果您处于内网环境,请确保网络连接配置正确。

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

验证是否成功安装

判断MindSpore Transformers是否安装成功可以执行以下代码:

python -c "import mindformers as mf;mf.run_check()"

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

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