【CAAI三期项目】mindspore运行时间过长,导致gpu内存不足
收藏回复举报
【CAAI三期项目】mindspore运行时间过长,导致gpu内存不足
t('forum.solved') 已解决
新人帖
发表于2023-06-05 12:24:17
0 查看

  在pytorch框架下的代码运行一个epoch大概是1分钟,迁移成mindspore后执行一个epoch用时25分钟左右。mindspore的程序长时间的运行使得实验室3090的卡在程序运行结束前就out of memory了。我考虑使用智启平台的卡,但是由于我迁移所用的mindspore为rc2.0版本,该平台已有镜像不支持该版本的安装,现在项目无法推进了。

  可否请求智启平台能否添加mindspore-rc2.0版本的镜像呢?

日志截图:

cke_8299.png

我要发帖子