张小白带你玩转Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4(14)尝试昇思跑LeNet-2
【摘要】 Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4+MindSpore 1.6.1
十四、在Ubuntu 18.04 X86上使用MindSpore CPU版训练LeNet网络
书接上回:上次张小白没有在200DK上完成LeNet网络训练,这让曾经能在 树莓派4B和Jetson Nano B01完成LeNet训练的张小白感到有些失望。
不过这也挡不住张小白试验LeNet的心,既然无法在200DK上做训练,那么用WSL2 ubuntu GPU环境(或者开发环境ubuntu )做LeNet训练,然后将训练好的结果送到200DK上做离线推理,应该是可以的吧?
先看下开发环境的环境:
有python3.7.5,但是没有mindspore,那就先在开发环境上安装MindSpore的CPU版本:
按照上面的选择来安装即可:
python3 -m pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/1.6.1/MindSpore/cpu/x86_64/mindspore-1.6.1-cp37-cp37m-linux_x86_64.whl --trusted-host ms-release.obs.cn-north-4.myhuaweicloud.com -i https://pypi.tuna.tsinghua.edu.cn/simple
下载mindpore的modelzoo代码仓:
准备数据集:
cd models/official/cv/lenet
mkdir MNIST
mkdir ckpt
cd MNIST
安装tree
查看数据集结构:
开始训练:
python3 train.py --data_path=./MNIST/ --device_target="CPU" --ckpt_path=./ckpt
。。。
。。。
训练结果在ckpt目录下:
执行评估:
python3 eval.py --data_path MNIST --ckpt_path ckpt/checkpoint_lenet-10_1875.ckpt --device_target="CPU"
导出MindIR模型:
python3 export.py --ckpt_file ckpt/checkpoint_lenet-10_1875.ckpt --file_format "MINDIR" --device_target="CPU"
执行完毕后,会生成一个lenet.mindir文件:
将lenet.mindir和ckpt文件拷贝到200DK上。
scp lenet.mindir HwHiAiUser@192.168.199.204:/home/HwHiAiUser/models/official/cv/lenet/
scp -r ckpt HwHiAiUser@192.168.199.204:/home/HwHiAiUser/models/official/cv/lenet/
下面来尝试使用训练好的模型在200DK上进行推理。
(未完待续)
【声明】本内容来自华为云开发者社区博主,不代表华为云及华为云开发者社区的观点和立场。转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息,否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱:
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
作者其他文章
张辉2022/04/27 22:09:301楼编辑删除举报
https://bbs.huaweicloud.com/blogs/345859
张小白带你玩转Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4(二)分设环境之运行环境准备
https://bbs.huaweicloud.com/blogs/345861
张小白带你玩转Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4(三)分设环境之运行环境准备dd
https://bbs.huaweicloud.com/blogs/345862
张小白带你玩转Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4(四)MindStudio安装配置
https://bbs.huaweicloud.com/blogs/345869
张小白带你玩转Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4(五)OpenCV的安装
https://bbs.huaweicloud.com/blogs/345905
张小白带你玩转Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4(六)番外篇:寻找失去的网卡
https://bbs.huaweicloud.com/blogs/345906
张小白带你玩转Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4(七)合设环境准备(mksd)
https://bbs.huaweicloud.com/blogs/346015
张小白带你玩转Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4(八)完成CANN进阶班大作业1
https://bbs.huaweicloud.com/blogs/347062
张小白带你玩转Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4(九)ATC模型转换
https://bbs.huaweicloud.com/blogs/348182
张辉2022/04/27 22:20:092楼编辑删除举报
https://bbs.huaweicloud.com/blogs/348248
张小白带你玩转Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4(11)番外:TF卡扩容
https://bbs.huaweicloud.com/blogs/349546
张小白带你玩转Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4(12)更好的TF卡扩容方式
https://bbs.huaweicloud.com/blogs/349547
张小白带你玩转Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4(13)尝试昇思跑LeNet-1
https://bbs.huaweicloud.com/blogs/349620
张小白带你玩转Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4(14)尝试昇思跑LeNet-2
https://bbs.huaweicloud.com/blogs/349621
张小白带你玩转Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4(15)尝试昇思跑LeNet-3
https://bbs.huaweicloud.com/blogs/349622
张小白带你玩转Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4(16)尝试昇思跑LeNet-4
https://bbs.huaweicloud.com/blogs/349626
张小白带你玩转Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4(17)玩转摄像头
https://bbs.huaweicloud.com/blogs/349654
韩令2022/06/15 05:31:123楼编辑删除举报
raise ValueError(f"For 'context.set_context', package type {__package_name__} support 'device_target' "
ValueError: For 'context.set_context', package type mindspore support 'device_target' type cpu, but got Ascend.
我看到https://toscode.gitee.com/mindspore/mindspore/issues/I4RF5X 这个论坛上说可能是因为安装的是mindspore cpu版本,但是训练的时候在context中指定了device_target为Ascend: 然后说是可以下载mindspore-ascend 1.3.0
我跟着您的帖子下载的是mindspore cpu 1.6.1版本的 但是您帖子里好像没有报错诶 请问是哪里要改一下代码吗 怎么在训练的时候指定device_target为cpu呢
张辉2022/06/15 06:09:35编辑删除举报
张辉2022/06/15 06:12:15编辑删除举报
韩令2022/06/15 06:18:20编辑删除举报