Installation

Confirming Version Matching Relationship

The currently supported hardware is the Atlas 800T A2 training server.

The current recommended Python version for the suite is 3.11.4.

MindSpore Transformers	MindSpore	CANN	Firmware & Drivers
1.5.0	2.6.0-rc1	8.1.RC1	25.0.RC1

Currently MindSpore Transformers recommends using a software package relationship as above.

Historical version matching relationship:

MindSpore Transformers	MindSpore	CANN	Firmware & Drivers
1.3.2	2.4.10	8.0.0	24.1.0
1.3.0	2.4.0	8.0.RC3.beta1	24.1.RC3
1.2.0	2.3.0	8.0.RC2.beta1	24.1.RC2

Install Firmware and Driver: Download the firmware and driver package through the Confirming Version Matching Relationship to download the installation package, and refer to the Ascend official tutorial - CANN Software Installation Guide for installation.
Install CANN and MindSpore: Use the officially provided Docker image (CANN, MindSpore are already included in the image, no need to install them manually) or follow the Manual Installation section on the MindSpore website for installation.

Currently only source code compilation installation is supported, users can execute the following command to install MindSpore Transformers:

git clone -v 1.5.0 https://gitee.com/mindspore/mindformers.git
cd mindformers
bash build.sh

To determine whether MindSpore Transformers has been successfully installed, execute the following code:

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

A similar result as below proves that the installation was successful:

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