安装指南
确认版本匹配关系
当前支持的硬件为 Atlas 800T A2、Atlas 800I A2、Atlas 900 A3 SuperPoD。
当前套件建议使用的Python版本为3.11.4。
MindSpore Transformers |
MindSpore |
CANN |
固件与驱动 |
---|---|---|---|
1.6.0 |
当前MindSpore Transformers建议使用如上的软件配套关系。
历史版本配套关系:
MindSpore Transformers |
MindSpore |
CANN |
固件与驱动 |
---|---|---|---|
1.5.0 |
|||
1.3.2 |
|||
1.3.0 |
|||
1.2.0 |
安装依赖软件
安装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!