Installation

View Source On Gitee

Version Matching Relationship

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

The current recommended Python version for the suite is 3.10.

MindFormers

MindSpore

CANN

Firmware & Drivers

Mirror Links

r1.3.0

2.4.0

CANN

Driver

TBD

Currently MindFormers recommends using a software package relationship as above.

Environment Installation

  1. Install Firmware and Driver: Download the firmware and driver package through the Version Matching Relationship to download the installation package, and refer to the Ascend official tutorial for installation.

  2. 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.

MindFormers Installation

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

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

Installation Verification

To determine whether MindFormers has been successfully installed, execute the following command:

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!