升级MindSpore和Mindformers版本后,无法使用NPU
收藏回复举报
升级MindSpore和Mindformers版本后,无法使用NPU
t('forum.solved') 已解决
新人帖
发表于2024-08-09 15:17:06
0 查看

环境配置

使用ModelArts的Notebook开发环境,系统原生镜像中软件版本为:

类型版本
MindSpore2.1.0
Mindformers1.0.0
CANN6.3.2
Python3.7
NPUAscend 910A
Driver23.0.rc2

在此镜像中,运行mindspore-quick-start.py任务是可以使用到NPU的。

但是当我尝试将系统环境升级到以下版本时,无法使用NPU进行简单的训练

类型版本
MindSpore2.3.0
Mindformers1.2.0
CANN8.0.RC2
Python3.10
NPUAscend 910A
Driver23.0.rc2
  • 升级Python 3.10:
conda deactivate
conda create -n new  python=3.10

conda activate new

python --version
# Python 3.10.14
wget https://ms-release.obs.cn-north-4.myhuaweicloud.com/2.3.0/MindSpore/unified/aarch64/mindspore-2.3.0-cp310-cp310-linux_aarch64.whl

pip install mindspore-2.3.0-cp310-cp310-linux_aarch64.whl
  • 升级Mindformers:从官方仓库下载并切换到r1.2.0分支
git clone https://gitee.com/mindspore/mindformers.git
cd mindformers

git tag
git checkout v1.2.0
build build.bash
wget cid:link_0
bash ./Ascend-cann-toolkit_8.0.RC2_linux-aarch64.run --install

# 更新ascend toolkit 环境变量
source Ascend/ascend-toolkit/set_env.sh

升级后环境如下: true

运行quick_start.py

python mindspore_quick_start.py

结果如下: true

并没有使用到NPU!

请问这种情况是什么原因呢?驱动版本过低导致的吗?

我要发帖子