AI Workbench 和 Brev 集成#

注意

这目前是 NVIDIA AI Workbench 的一项实验性功能，供现有 Brev 用户测试并提供反馈。此功能需要 Brev CLI。请确保您的系统（Mac/Linux）或 AI Workbench 发行版（Windows）上安装了 Brev CLI。

此功能允许您将现有的 Brev 实例配置为 Workbench 位置，而无需进行通常的手动安装和配置。有关更多信息，请参阅 AI Workbench 位置。

请使用本文档了解以下内容

先决条件#

在将 Brev 实例与 AI Workbench 一起使用之前，您必须具备以下条件

  • 本地安装的 AI Workbench。

  • Brev 帐户。您可以使用 Google 或 GitHub 凭据注册或登录到 Brev。

  • 本地安装的最新版本 Brev CLI。

如果未安装 Brev CLI，您可以从命令行安装它。

有关 Mac 和 Ubuntu 的命令，请参阅 Brev 文档。

注意

在 Windows 上，您需要使用 sudo 权限将 Brev CLI 安装到 NVIDIA AI Workbench WSL 发行版 NVIDIA-Workbench 中。

但是，sudo 权限在发行版中被阻止，您必须从 Windows shell 或 Powershell 以 root 身份输入它。

您可以使用以下命令将 Brev CLI 安装到 NVIDIA-Workbench WSL 发行版中

1wsl -d NVIDIA-Workbench -u root -- bash -c "$(curl -fsSL https://raw.githubusercontent.com/brevdev/brev-cli/main/bin/install-latest.sh)"

在 Brev 上获取云实例#

使用以下步骤在 Brev 上获取云实例。

  1. 登录到 Brev 控制台。

  2. 打开实例选项卡（如果尚未打开）。

  3. 选择新建。

  4. 选择 VM 模式。

    注意

    无需预安装 Jupyter。

  5. 打开云选项卡。

  6. 在提供商下，选择一个提供商，例如 GCP。这将使用该提供商中可用的 GPU 列表更新 GPU 列表。

  7. 从水平栏中选择 GPU 类型。这将列出可用的 GPU，例如 NVIDIA A10。

  8. 从垂直列表中选择所需的配置。这将列出该提供商上该 GPU 类型的不同 GPU 实例配置。

  9. 在实例名称下，为您的实例提供一个简短的名称，例如 My-A10-instance。

  10. 选择部署。这将在 Brev 中启动您的实例。您可能需要等待几分钟才能启动实例。

使用集成配置您的 Brev 实例作为位置#

在 Brev 中拥有可用的云实例后，您可以使用 AI Workbench 与 Brev 的集成将云实例添加为 AI Workbench 位置。

  1. 通过执行以下操作之一打开命令 shell

    • 在 macOS 或 Ubuntu 上 — 打开终端。

    • 在 Windows 上 — 通过从 Windows 终端应用程序中选择 NVIDIA-Workbench WSL 终端来打开它。当终端打开时，您应该看到类似于以下的命令提示符

    1workbench@computer-name:~$

    提示

    您也可以使用 Windows 命令提示符或 Windows PowerShell 通过命令 wsl -d NVIDIA-Workbench 访问发行版。

  2. 登录到您的 Brev 帐户。

    1brev login

    按照提供的说明使用浏览器链接或提供的 CLI 命令链接登录。

  3. 如果您有多个组织，请将 Brev 指向相应的组织。

    1brev set <org-name>

    例如，如果 j_smith-hq 是您的组织名称，您将输入

    1brev set j_smith-hq

  4. 要将您的 Brev 实例配置为位置，请运行以下命令。

    1nvwb create context --brev-instance-name <my-brev-instance>

    当提示时，选择是以继续连接。

  5. 要查看您的可用位置列表中的新 brev 实例，请运行以下代码。

    1nvwb list contexts

    输出应类似于以下内容。

    1        NAME             |        DESCRIPTION          |     HOSTNAME     |      STATUS     | EXTERNAL ACCESS
2-------------------------|-----------------------------|------------------|-----------------|----------------------
3brev-<my-brev-instance>  | Brev managed instance       | <host-ip>        | Not Running     |   Not Enabled
4                         | <gpu-name> (gpu)            |                  |                 |
5local                    | My Computer                 | localhost        | Not Runnng      |   Not Enabled

  6. 要激活您的 Brev 实例，请运行以下代码。

    1nvwb activate <my-brev-instance>

将您的 Brev 实例与 AI Workbench 一起使用#

配置 Brev 实例后，您可以使用 AI Workbench CLI 或 AI Workbench 桌面应用程序来使用它。有关更多信息，请参阅基本快速入门和基本快速入门 (CLI)。

在桌面应用程序中，选择您的 Brev 位置，类似于任何本地或远程位置。您的 Brev 实例在 AI Workbench 中以 brev 为前缀显示名称。

在本页中