AI开发平台MODELARTS-Step2 准备脚本文件并上传至OBS中:训练脚本mindspore-verification.py文件

时间：2024-04-30 17:31:41

AI开发平台MODELARTS

训练脚本mindspore-verification.py文件

mindspore-verification.py文件内容如下：

import os
import numpy as np
from mindspore import Tensor
import mindspore.ops as ops
import mindspore.context as context

print('Ascend Envs')
print('------')
print('JOB_ID: ', os.environ['JOB_ID'])
print('RANK_TABLE_FILE: ', os.environ['RANK_TABLE_FILE'])
print('RANK_SIZE: ', os.environ['RANK_SIZE'])
print('ASCEND_DEVICE_ID: ', os.environ['ASCEND_DEVICE_ID'])
print('DEVICE_ID: ', os.environ['DEVICE_ID'])
print('RANK_ID: ', os.environ['RANK_ID'])
print('------')

context.set_context(device_target="Ascend")
x = Tensor(np.ones([1,3,3,4]).astype(np.float32))
y = Tensor(np.ones([1,3,3,4]).astype(np.float32))

print(ops.add(x, y))

上一篇：AI开发平台MODELARTS-使用基础镜像构建新的训练镜像:基于训练基础镜像构建新镜像的操作步骤

下一篇：AI开发平台MODELARTS-场景描述:场景描述