MindSpore的基础功能

配置运行信息

MindSpore通过context.set_context来配置运行需要的信息，如运行模式、后端信息、硬件等信息。

导入context模块，配置运行需要的信息。

代码如下：

import os

import argparse

from mindspore import context

parser = argparse.ArgumentParser(description='MindSpore LeNet Example')

parser.add_argument('--device_target', type=str, default="CPU", choices=['Ascend', 'GPU', 'CPU'])

args = parser.parse_known_args()[0]