添加 conda 环境

本页面介绍了如何将 conda 环境添加到 Vertex AI Workbench 实例。

概览

当您将 conda 环境添加到 Vertex AI Workbench 实例时,它会在实例的 JupyterLab 界面中显示为内核

您可以向 Vertex AI Workbench 实例添加 conda 环境,以使用 Vertex AI Workbench 实例中没有的内核。例如,您可以为 R 和 Apache Beam 添加 conda 环境。或者,您可以为特定旧版可用框架(例如 TensorFlow、PyTorch 或 Python)添加 conda 环境。

须知事项

创建 Vertex AI Workbench 实例(如果您尚未创建)。

打开 JupyterLab

  1. 在 Google Cloud 控制台中,进入实例页面。

    转到实例

  2. 在 Vertex AI Workbench 实例名称旁边,点击打开 JupyterLab

    您的 Vertex AI Workbench 实例会打开 JupyterLab。

添加 conda 环境

您可以通过在实例的 JupyterLab 终端中输入命令来添加 conda 环境。

  1. 在 JupyterLab 中,选择文件 > 新建 > 终端

  2. 终端窗口中,输入以下命令:

    # Creates a conda environment.
    conda create -n CONDA_ENVIRONMENT_NAME -y
    conda activate CONDA_ENVIRONMENT_NAME
    
    # Install packages using a pip local to the conda environment.
    conda install pip
    pip install PACKAGE
    
    # Adds the conda kernel.
    DL_ANACONDA_ENV_HOME="${DL_ANACONDA_HOME}/envs/CONDA_ENVIRONMENT_NAME"
    python -m ipykernel install --prefix "${DL_ANACONDA_ENV_HOME}" --name CONDA_ENVIRONMENT_NAME --display-name KERNEL_DISPLAY_NAME

    替换以下内容:

    • CONDA_ENVIRONMENT_NAME:您为环境选择的名称
    • PACKAGE:您要安装的软件包
    • KERNEL_DISPLAY_NAME:JupyterLab 界面中内核图块的显示名称
  3. 在安装到给定 conda 环境时,可以创建默认内核。您可以使用以下命令移除默认内核:

    rm -rf "/opt/micromamba/envs/CONDA_ENVIRONMENT_NAME/share/jupyter/kernels/python3
  4. 如需查看新内核,请执行以下操作:

    1. 刷新页面。

    2. 选择 File > New Launcher

    该内核列在启动器窗口中,与其他内核一起列出。

默认情况下,conda 可能会使用系统 pip 文件夹(例如 /usr/bin/pip)中的 pip 软件包。运行 conda install pip 可确保设置使用环境的本地 pip。

安装示例:R Essentials

以下示例在名为 r 的 conda 环境中安装 R Essentials。

conda create -n r
conda activate r
conda install -c r r-essentials

安装示例:pip 软件包

以下示例通过 requirements.txt 文件安装 pip 软件包。

conda create -n myenv
conda activate myenv
conda install pip
pip install -r requirements.txt
DL_ANACONDA_ENV_HOME="${DL_ANACONDA_HOME}/envs/myenv"
python -m ipykernel install --prefix "${DL_ANACONDA_ENV_HOME}" --name myenv --display-name myenv

问题排查

如需诊断和解决与添加 conda 环境相关的问题,请参阅排查 Vertex AI Workbench 问题

后续步骤