在modelarts中创建mindspore环境,并安装mindvision,并进行深度学习训练(华为)

1.背景:

华为在各大重点大学,都有基于mindspore的深度学习课程,常常会发放代金券供学生们实验并提交作业(我白嫖了NKU的)。一开始我也尝试过colab或者本机PC(windows),但是发现这些环境mindspore环境要么支持地不太好(例如一些mindspore配套的包不能适配),要么就是自己手动安装mindspore过程中,还需要解决一些包冲突问题。总而言之,既然发了代金券,不用白不用。modelarts上实验大体记录:
在modelarts中创建mindspore环境,并安装mindvision,并进行深度学习训练(华为)_第1张图片
实验代码(华为老师布置的作业):
生成式对抗网络 生成二次元头像

2.实验场景:

modelartsNotebook环境下训练生成式对抗网络 生成二次元头像

mindspore官网安装指导
如下:获取安装命令
在modelarts中创建mindspore环境,并安装mindvision,并进行深度学习训练(华为)_第2张图片

注:不推荐在PC(windows)上安装mindspore进行训练,推荐去 modelartsNotebook环境下训练,这也是本文的教程主要正对针对的场景在modelarts中创建mindspore环境,并安装mindvision,并进行深度学习训练(华为)_第3张图片

注:截止2022年4月26日,mindspore对windows环境的支持依旧不太好,仅有windows×86 cpu的少数几个版本可选,而mindspore自研的mindvision包(一个视觉包,很多情况下都是要使用的),仅支持gpu版本,会因此而做无用功

modelarts notebook,mindspore1.5版本,DCGAN训练结果:
在modelarts中创建mindspore环境,并安装mindvision,并进行深度学习训练(华为)_第4张图片
在modelarts中创建mindspore环境,并安装mindvision,并进行深度学习训练(华为)_第5张图片


在colab上也能训练,但是,如果分到了Tesla K80(单核2线程)
那么在安装环境后,还需要:设置num_parallel_workers=1(不然会报错,原因是示例中代码开了多线程(2个以上))
训完结果也是惨不忍睹:
在modelarts中创建mindspore环境,并安装mindvision,并进行深度学习训练(华为)_第6张图片
在modelarts中创建mindspore环境,并安装mindvision,并进行深度学习训练(华为)_第7张图片

你可能感兴趣的:(深度学习,华为,华为云,tensorflow)