机器学习系统:设计和实现
上QQ阅读APP看书,第一时间看更新

2.2.1 环境配置

MindSpore通过context.set_context配置运行需要的信息,如运行模式、后端信息、硬件信息等。代码2.1导入context模块,配置运行需要的信息。此代码将持续更新。

代码2.1 配置运行信息

上述配置样例(1)运行使用图模式。根据实际情况配置硬件信息,譬如代码运行在Ascend AI处理器上,则device_target选择Ascend,代码运行在CPU、GPU上同理。