你当前正在访问 Microsoft Azure Global Edition 技术文档网站。 如果需要访问由世纪互联运营的 Microsoft Azure 中国技术文档网站,请访问 https://docs.azure.cn

为 Azure 机器学习设置 Python 开发环境

适用范围:Python SDK azure-ai-ml v2(最新版)

了解如何为 Azure 机器学习配置 Python 开发环境。

下表描述了本文所述的每个开发环境及其优点和缺点。

环境 优点 缺点
本地环境 可以全面控制开发环境和依赖项。 使用所选的任何生成工具、环境或 IDE 来运行。 入门需要更长的时间。 必须安装必要的 SDK 包,此外,必须安装一个环境(如果尚未安装)。
Data Science Virtual Machine (DSVM) 类似于基于云的计算实例(已预装 Python),但预装了其他常用的数据科学和机器学习工具。 易于缩放,并可与其他自定义工具和工作流结合使用。 与基于云的计算实例相比,入门过程更慢。
Azure 机器学习计算实例 最容易入门。 SDK 已安装在工作区 VM 中,笔记本教程已预先克隆,随时可供运行。 缺少对开发环境和依赖项的控制。 Linux VM 会产生额外的成本(可以停止不使用的 VM,以免产生费用)。 请参阅定价详细信息

本文还将提供以下工具的更多用法提示:

  • Jupyter Notebook:如果已在使用 Jupyter Notebook,则应安装 SDK 的某些附加功能。

  • Visual Studio Code:如果使用 Visual Studio Code,Azure 机器学习扩展包含对 Python 的语言支持,还包含便于更方便、更高效地使用 Azure 机器学习的功能。

先决条件

仅限本地和 DSVM:创建一个工作区配置文件

工作区配置文件是一个 JSON 文件,用于告知 SDK 如何与 Azure 机器学习工作区进行通信。 该文件命名为 config.json,其格式如下:

{
    "subscription_id": "<subscription-id>",
    "resource_group": "<resource-group>",
    "workspace_name": "<workspace-name>"
}

此 JSON 文件必须采用包含 Python 脚本或 Jupyter Notebook 的目录结构。 它可以位于同一目录(名为 .azureml* 的子目录)中,也可以位于父目录中。

若要从代码使用此文件,请使用 MLClient.from_config 方法。 此代码从文件中加载信息,并连接到工作区。

使用下列方法之一创建工作区配置文件:

  • Azure 机器学习工作室

    下载文件

    1. 登录到 Azure 机器学习工作室
    2. 在右上方的 Azure 机器学习工作室工具栏中,选择你的工作区名称。
    3. 选择“下载配置文件”链接。

    屏幕截图显示如何下载配置文件。

  • Azure 机器学习 Python SDK

    创建脚本以连接到你的 Azure 机器学习工作区。 确保将 subscription_idresource_groupworkspace_name 替换为你自己的值。

    适用于:Python SDK azure-ai-ml v2(当前版本)

    #import required libraries
    from azure.ai.ml import MLClient
    from azure.identity import DefaultAzureCredential
    
    #Enter details of your Azure Machine Learning workspace
    subscription_id = '<SUBSCRIPTION_ID>'
    resource_group = '<RESOURCE_GROUP>'
    workspace = '<AZUREML_WORKSPACE_NAME>'
    
    #connect to the workspace
    ml_client = MLClient(DefaultAzureCredential(), subscription_id, resource_group, workspace)
    

本地计算机或远程 VM 环境

你可以在本地计算机或远程虚拟机上设置环境,例如 Azure 机器学习计算实例或 Data Science VM。

若要配置本地开发环境或远程 VM,请执行以下操作:

  1. 创建 Python 虚拟环境(virtualenv,conda)。

    注意

    建议使用 AnacondaMiniconda 来管理 Python 虚拟环境并安装包,虽然这不是必需的。

    重要

    如果在 Linux 或 macOS 上操作,并使用除 bash 以外的 shell(例如 zsh),则在运行某些命令时可能会收到错误消息。 若要解决此问题,请使用 bash 命令启动新的 bash shell,然后运行命令。

  2. 激活新创建的 Python 虚拟环境。

  3. 安装 Azure 机器学习 Python SDK

  4. 若要将本地环境配置为使用你的 Azure 机器学习工作区,请创建一个工作区配置文件或使用现有文件。

设置本地环境后,便可以开始使用 Azure 机器学习。 若要开始使用,请参阅教程:Azure 机器学习一天导览

Jupyter Notebook

运行本地 Jupyter Notebook 服务器时,建议为你的 Python 虚拟环境创建一个 IPython 内核。 这有助于确保实现预期的内核和包导入行为。

  1. 启用环境特定的 IPython 内核

    conda install notebook ipykernel
    
  2. 为你的 Python 虚拟环境创建一个内核。 请确保将 <myenv> 替换为你的 Python 虚拟环境的名称。

    ipython kernel install --user --name <myenv> --display-name "Python (myenv)"
    
  3. 启动 Jupyter Notebook 服务器

提示

有关示例笔记本,请参阅 AzureML-Examples 存储库。 SDK 示例位于 /sdk/python 下。 例如,配置笔记本示例。

Visual Studio Code

若要使用 Visual Studio Code 进行开发:

  1. 安装 Visual Studio Code
  2. 安装 Azure 机器学习 Visual Studio Code 扩展(预览版)。

安装 Visual Studio Code 扩展后,请使用它来执行以下操作:

Azure 机器学习计算实例

Azure 机器学习计算实例是一个安全的基于云的 Azure 工作站,为数据科学家提供 Jupyter Notebook 服务器、JupyterLab 和一个完全托管的机器学习环境。

无需为计算实例安装或配置任何组件。

随时可从 Azure 机器学习工作区内部创建组件。 只需提供名称并指定 Azure VM 类型即可。 通过创建帮助入门的资源立即开始试用。

若要详细了解计算实例(包括如何安装包),请参阅创建 Azure 机器学习计算实例

提示

若要防止未使用的计算实例产生费用,请启用空闲关闭

除了 Jupyter Notebook 服务器和 JupyterLab 以外,还可以在 Azure 机器学习工作室内的集成笔记本功能中使用计算实例。

还可以使用 Azure 机器学习 Visual Studio Code 扩展以通过 VS Code 连接到远程计算实例

数据科学虚拟机

Data Science VM 是一种可用作开发环境的自定义虚拟机 (VM) 映像。 它专为数据科学工作而设计,其中预配置了工具和软件,例如:

  • TensorFlow、PyTorch、Scikit-learn、XGBoost 和 Azure 机器学习 SDK 等包
  • Spark Standalone 和 Drill 等常用数据科学工具
  • Azure CLI、AzCopy 和存储资源管理器等 Azure 工具
  • Visual Studio Code 和 PyCharm 等集成开发环境 (IDE)
  • Jupyter Notebook 服务器

有关更全面的工具列表,请参阅 Data Science VM 工具指南

重要

如果你计划将 Data Science VM 用作训练或推理作业的计算目标,则仅 Ubuntu 受支持。

若要使用 Data Science VM 作为开发环境,请执行以下操作:

  1. 使用下列方法之一创建一个 Data Science VM:

    • 使用 Azure 门户创建一个 UbuntuWindows DSVM。

    • 使用 ARM 模板创建 Data Science VM

    • 使用 Azure CLI

      若要创建 Ubuntu Data Science VM,请使用以下命令:

      # create a Ubuntu Data Science VM in your resource group
      # note you need to be at least a contributor to the resource group in order to execute this command successfully
      # If you need to create a new resource group use: "az group create --name YOUR-RESOURCE-GROUP-NAME --location YOUR-REGION (For example: westus2)"
      az vm create --resource-group YOUR-RESOURCE-GROUP-NAME --name YOUR-VM-NAME --image microsoft-dsvm:linux-data-science-vm-ubuntu:linuxdsvmubuntu:latest --admin-username YOUR-USERNAME --admin-password YOUR-PASSWORD --generate-ssh-keys --authentication-type password
      

      若要创建 Windows DSVM,请使用以下命令:

      # create a Windows Server 2016 DSVM in your resource group
      # note you need to be at least a contributor to the resource group in order to execute this command successfully
      az vm create --resource-group YOUR-RESOURCE-GROUP-NAME --name YOUR-VM-NAME --image microsoft-dsvm:dsvm-windows:server-2016:latest --admin-username YOUR-USERNAME --admin-password YOUR-PASSWORD --authentication-type password
      
  2. 为 Azure 机器学习 SDK 创建 conda 环境:

    conda create -n py310 python=310
    
  3. 创建环境后,激活它并安装 SDK

    conda activate py310
    pip install azure-ai-ml azure-identity
    
  4. 若要将 Data Science VM 配置为使用你的 Azure 机器学习工作区,请创建一个工作区配置文件或使用现有的工作区配置文件。

    提示

    你可以使用 Visual Studio Code 和 Azure 机器学习 Visual Studio Code 扩展(与本地环境类似)与 Azure 机器学习进行交互。

    有关详细信息,请参阅 Data Science Virtual Machine

后续步骤