ModelArts模型导入自定义配置深度学习框架版本

举报
星月菩提 发表于 2020/11/24 17:27:59 2020/11/24
【摘要】 以pytorch 1.5和tensorflow 1.14为例,如何在模型导入中进行配置pytorch 1.5和tensorflow 1.14。

当前ModelArts各功能都只支持有限的框架版本,如果想要使用一个预置框架没有的版本应该如何处理?

下面以pytorch 1.5和tensorflow 1.14为例,如何在模型导入中进行配置pytorch 1.5和tensorflow 1.14。

Pytorch 1.5要基于cuda 10.1版本以上,tensorflow 1.14需要cuda10.

Torch 安装命令参考


https://pytorch.org/get-started/previous-versions/


 

模型导入是根据config.json进行配置环境的,配置文件编写说明

https://support.huaweicloud.com/engineers-modelarts/modelarts_23_0092.html

runtime字段指定了使用的docker基础镜像,可选的范围请看链接


dependencies字段指定环境依赖python包信息

1      Pytorch 1.5

Runtime字段配置为

dependencies字段配置为


2      Tensorflow 1.14

2.1        Gpu

Runtime字段配置为


dependencies字段配置为



2.2        Cpu

Runtime字段配置为

dependencies字段配置为




【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。