AI开发平台MODELARTS-从0制作自定义镜像用于创建训练作业（PyTorch+CPU/GPU）:Step2 准备训练脚本并上传至OBS

时间：2024-08-16 20:39:00

AI开发平台MODELARTS

Step2 准备训练脚本并上传至OBS

准备本案例所需的训练脚本“pytorch-verification.py”文件，并上传至OBS桶的“obs://test-modelarts/pytorch/demo-code/”文件夹下。

“pytorch-verification.py”文件内容如下：

import torch
import torch.nn as nn

x = torch.randn(5, 3)
print(x)

available_dev = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
y = torch.randn(5, 3).to(available_dev)
print(y)

上一篇：AI开发平台MODELARTS-从0制作自定义镜像用于创建训练作业（PyTorch+CPU/GPU）:Step1 创建OBS桶和文件夹

下一篇：AI开发平台MODELARTS-从0制作自定义镜像用于创建训练作业（PyTorch+CPU/GPU）:Step1 创建OBS桶和文件夹

AI开发平台MODELARTS-从0制作自定义镜像用于创建训练作业（PyTorch+CPU/GPU）:Step2 准备训练脚本并上传至OBS

意见反馈

0/200

提交取消

提交成功！非常感谢您的反馈，我们会继续努力做到更好反馈提交失败！请稍后重试！

AI开发平台MODELARTS-从0制作自定义镜像用于创建训练作业（PyTorch+CPU/GPU）:Step2 准备训练脚本并上传至OBS

AI开发平台MODELARTS-从0制作自定义镜像用于创建训练作业（PyTorch+CPU/GPU）:Step2 准备训练脚本并上传至OBS

Step2 准备训练脚本并上传至OBS

7*24

备案

专业服务

退订

建议反馈

售前咨询热线