什么是 NVIDIA AI Workbench?#

NVIDIA AI Workbench 是一个面向数据科学、机器学习和 AI 项目开发的开发者工具包。AI Workbench 是免费的,您可以在几分钟内将其安装在本地或远程计算机上,它同时提供桌面应用程序和命令行界面 (CLI)。

AI Workbench 实现了以下功能:

  • 在您的笔记本电脑上开发,然后将工作负载转移到数据中心或云中的可扩展 GPU 资源。

  • 通过 GitHubGitLab 协作和分发您的工作。

  • JupyterLabVS Code 或文本编辑器中编写和运行代码。

  • 从 GitHub 克隆功能齐全的 NVIDIA 提供的示例,用于执行检索增强生成 (RAG)、模型微调和 RAPIDS 数据科学工作流等任务。

  • 使用 NVIDIA 的预构建容器 或您自己的 完全自定义容器,在容器化环境中进行开发。

../_images/deploy.png

AI Workbench 在以下操作系统上受支持:

  • Windows 11(内部版本 22000 或更高版本)、Windows 10(内部版本 19044 或更高版本)

  • Ubuntu 24.04、Ubuntu 22.04

  • macOS Monterrey (12) 或更高版本

AI Workbench 组件#

AI Workbench 由以下组件组成:

  • 桌面应用程序 — AI Workbench 桌面应用程序。一个跨平台应用程序,为 AI Workbench 提供图形用户界面 (GUI)。

  • 命令行界面 (CLI) — AI Workbench CLI。面向喜欢在终端中工作或想要编写 AI Workbench 流程脚本的用户的命令行界面。

  • 服务/服务器 — AI Workbench 服务。核心组件,提供用于管理项目和与之交互的 API,并处理后端操作和管理任务。

  • 凭据管理器 — AI Workbench 凭据管理器。一个小型应用程序,与您主机的密钥链或密钥存储集成。

有关更多信息,请参阅 AI Workbench 系统

位置#

安装了 AI Workbench 的计算机称为位置。您安装 AI Workbench 的第一个位置是您的本地计算机,它可以是仅 CPU 的计算机。稍后,您可以在一台或多台远程计算机上安装 AI Workbench,例如工作站、服务器、云实例和虚拟机。有关更多信息,请参阅 AI Workbench 位置

提示

您可以使用 Brev 实例作为位置。有关更多信息,请参阅 AI Workbench 和 Brev 集成

项目和容器#

AI Workbench 项目包含您完成工作所需的一切,例如代码、数据、模型、环境配置、元数据和工作历史记录。AI Workbench 项目是特殊格式的 Git 存储库,AI Workbench 使用该格式和元数据为容器和版本管理提供自动化。有关更多信息,请参阅 AI Workbench 项目

项目的环境是 AI Workbench 构建和运行的容器。NVIDIA 提供默认容器,您可以从中选择作为每个新项目的起点。有关更多信息,请参阅 AI Workbench 项目的容器环境

NVIDIA 提供的默认容器包括以下内容:

  • 用于 AI Workbench 的 PyTorch

  • 用于 AI Workbench 的基本 Python

  • 带有 CUDA 11.7、12.0 和 12.2 的 Python

  • 带有 CUDA 12.0 的 RAPIDS。

集成#

AI Workbench 允许您通过个人访问令牌 (PAT) 和 Oauth 集成等身份验证方法连接到外部系统,例如容器注册表和 Git 服务器。AI Workbench 安全地存储您的凭据,因此您不必每次都登录。在您通过身份验证连接到外部系统后,您可以从在任何位置(包括远程位置)运行的任何项目访问它。有关更多信息,请参阅 AI Workbench 集成

应用程序#

应用程序是在 AI Workbench 项目中或与 AI Workbench 项目一起使用的软件或用户界面。例如:

  • JupyterLab — 每个 NVIDIA 提供的基本容器选项都默认安装了 JupyterLab。

  • Visual Studio Code — 如果您的本地计算机上安装了 VS Code,则可以将其连接到您的 Workbench 项目。

  • Gradio 应用程序 — 一些项目安装了简单的 Web 应用程序,例如聊天应用程序或 RAG 应用程序。

有关更多信息,请参阅 AI Workbench 应用程序

下一步#

另请参阅#