**在 Azure Machine Learning 上运行 Experiment** === NNI 支持在 [AML](https://azure.microsoft.com/zh-cn/services/machine-learning/) 上运行 Experiment,称为 aml 模式。 ## 设置环境 步骤 1. 参考[指南](../Tutorial/QuickStart.md)安装 NNI。 步骤 2. 按照[文档](https://docs.microsoft.com/zh-cn/azure/machine-learning/how-to-manage-workspace-cli),创建 AML 账户。 步骤 3. 获取账户信息。 ![](../../img/aml_account.png) 步骤 4. 安装 AML 包环境。 ``` python3 -m pip install azureml --user python3 -m pip install azureml-sdk --user ``` ## 运行 Experiment 以 `examples/trials/mnist-tfv1` 为例。 NNI 的 YAML 配置文件如下: ```yaml authorName: default experimentName: example_mnist trialConcurrency: 1 maxExecDuration: 1h maxTrialNum: 10 trainingServicePlatform: aml searchSpacePath: search_space.json #可选项: true, false useAnnotation: false tuner: #可选项: TPE, Random, Anneal, Evolution, BatchTuner, MetisTuner, GPTuner #SMAC (SMAC should be installed through nnictl) builtinTunerName: TPE classArgs: #可选项: maximize, minimize optimize_mode: maximize trial: command: python3 mnist.py codeDir: . computeTarget: ${replace_to_your_computeTarget} image: msranni/nni amlConfig: subscriptionId: ${replace_to_your_subscriptionId} resourceGroup: ${replace_to_your_resourceGroup} workspaceName: ${replace_to_your_workspaceName} ``` 注意:如果用 aml 模式运行,需要在 YAML 文件中设置 `trainingServicePlatform: aml`。 与[本机模式](LocalMode.md)的 Trial 配置相比,aml 模式下的键值还有: * computeTarget * 必填。 要在 AML 工作区中使用的计算机集群名称。 * image * 必填。 作业中使用的 Docker 映像名称。 amlConfig: * subscriptionId * Azure 订阅的 Id * resourceGroup * 账户的资源组 * workspaceName * 账户的工作区名称