AI开发平台MODELARTS-从0制作自定义镜像用于创建训练作业（MindSpore+Ascend）:训练mindspore-verification.py文件

时间：2024-08-16 20:38:58

训练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-从0制作自定义镜像用于创建训练作业（MindSpore+Ascend）:Step2 准备脚本文件并上传至OBS中

下一篇：AI开发平台MODELARTS-从0制作自定义镜像用于创建训练作业（MindSpore+Ascend）:Step2 准备脚本文件并上传至OBS中