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

时间：2025-03-12 09:40: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 到 1 制作自定义镜像并用于训练（PyTorch+CPU/GPU）:Step5 上传镜像至SWR服务

下一篇：AI开发平台MODELARTS-示例：从 0 到 1 制作自定义镜像并用于训练（PyTorch+CPU/GPU）:Step5 上传镜像至SWR服务

新客秒杀 L实例 2核1G 2M

29元/年

企业专享 X实例 2核4G 5M

198元/年

域名建站 com域名 1元

立即前往

免费体验 90+云产品免费体验

立即前往

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

意见反馈

0/200

提交取消

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