ModelArts模型导入自定义配置深度学习框架版本
【摘要】 以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)