管理 conda 环境

本页介绍了如何在 Vertex AI Workbench 实例中管理 conda 环境。

概览

如果您已向 Vertex AI Workbench 实例添加 conda 环境，该环境会在实例的 JupyterLab 界面中显示为kernel。

您可能向实例添加了 conda 环境，以使用默认 Vertex AI Workbench 实例中没有的内核。本页介绍了如何修改和删除该内核。

打开 JupyterLab

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

    转到实例

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

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

修改 conda 内核

Vertex AI Workbench 实例预安装了 PyTorch 和 TensorFlow 等框架。如果需要其他版本，可以在相关的 conda 环境中使用 pip 来修改库。

例如，如果要升级 PyTorch：

# Check the name of the conda environment for PyTorch
conda env list

# Activate the environment for PyTorch
conda activate pytorch

# Display the PyTorch version
python -c "import torch; print(torch.__version__)"

# Make sure to use pip from the conda environment for PyTorch
# This should be `/opt/conda/envs/pytorch/bin/pip`
which pip

# Upgrade PyTorch
pip install --upgrade torch

删除 conda 内核

某些 conda 软件包会在安装时向您的环境中添加默认内核。例如，当您安装 R 时，conda 可能还会添加一个 python3 内核。这可能会导致您的环境中的内核重复。为避免内核重复，请先删除默认内核，然后再创建具有相同名称的新内核。

rm -rf /opt/conda/envs/CONDA_ENVIRONMENT_NAME/share/jupyter/kernels/python3

问题排查

如需诊断和解决与在 Vertex AI Workbench 实例中管理 conda 环境相关的问题，请参阅排查 Vertex AI Workbench 问题。

后续步骤