演练:自定义您的环境 (CLI)#

使用本文档学习如何通过向您的 NVIDIA AI Workbench 项目添加软件包来自定义您的容器环境。有关完整列表,请参阅快速入门和演练

您在创建 AI Workbench 项目时选择的基础容器可能不具备您所需的所有功能。您可以通过安装其他软件包、自定义挂载和定义新的环境变量来自定义您的项目。

在本演练中,您将使用 AI Workbench CLI 执行以下任务

  1. 启动 AI Workbench 并打开项目

  2. 通过安装软件包自定义您的项目

  3. 重建您的项目环境

  4. 通过在代码中使用新软件包进行测试

注意

要创建您自己的完全自定义容器,请参阅高级演练:使用您自己的容器

先决条件#

在完成本演练中的步骤之前,您需要以下各项

启动 AI Workbench 并打开项目#

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

    • 在 macOS 或 Ubuntu 上 — 打开命令 shell。

    • 在 Windows 上 — 通过在您的应用程序列表中搜索 WSL 打开 WSL 应用程序。当 WSL 应用程序打开时,您应该看到类似于以下的命令提示符

      1workbench@computer-name:~$
      

      提示

      您也可以使用终端、命令提示符或 Windows PowerShell,并通过使用命令 wsl -d NVIDIA-Workbench 访问 WSL。

  2. 通过运行以下命令检查您计算机上可用的位置(上下文)。

    1nvwb list contexts
    

    如果您尚未添加任何远程位置,则唯一可用的位置是本地计算机。输出应类似于以下内容。

    1  NAME  | DESCRIPTION | HOSTNAME  |  STATUS
    2--------|-------------|-----------|---------------
    3  local | My Computer | localhost | Not Running
    
  3. 运行以下命令以启动本地位置。

    1nvwb activate local
    

    AI Workbench 启动并连接到您的计算机,并且 (nvwb:local) 现在出现在您的命令提示符的开头。

  4. 运行以下命令以查看您的项目的名称和状态。

    1nvwb list projects
    
  5. 运行以下命令以打开您的项目。将 <您的项目名称> 更改为您的项目的名称。

    1nvwb open <your project name>
    

    AI Workbench 打开您的项目,更改为项目目录,并且 (nvwb:local/<您的项目名称>) 现在出现在您的命令提示符的开头。

通过安装软件包自定义您的项目#

使用以下步骤向您的 AI Workbench 项目添加软件包。

  1. 运行以下命令以打开 JupyterLab。

    JupyterLab 在您的默认浏览器中打开,您可以开始工作了。

    1nvwb start jupyterlab
    

    JupyterLab 在您的默认浏览器中打开,您可以开始工作了。

  2. 在 JupyterLab 中,在文件浏览器中,双击 requirements.txt 以打开它。

    该文件打开后,您应该看到 jupyterlab 的条目。

  3. 移动到文件中的第一个空行,键入 numpy,然后保存文件。

  4. 在 JupyterLab 中,选择文件,然后单击关闭。确认关闭并关闭浏览器选项卡。

重建您的项目环境#

每当您更改项目环境时,您必须重建项目才能访问更改。使用以下步骤重建您的环境。

  1. 在您的命令 shell 中,通过运行以下命令检查您的项目的状态。

    1nvwb status
    

    您应该看到类似于以下的输出。

    1Name: test-cli-proj (/home/workbench/nvidia-workbench/test-cli-proj)
    2Description: A hello world test project
    3Environment: Full Build Required
    4Container: Running
    5Applications:
    6  jupyterlab: Stopped
    7Remote Repository: https://github.com/nkmcalli/test-cli-proj.git (Up-to-date)
    8Local Repository: 1 Modified
    

    第 8 行指示您通过编辑 requirements.txt 文件更改了您的项目。第 3 行指示您所做的更改需要您重建您的项目。

  2. 运行以下命令以停止您的容器环境。如果容器已停止,您将看到一条消息,指示容器未运行。继续下一步。

    1nvwb stop --container
    
  3. 运行以下命令以重建您的项目。

    1nvwb build
    

    项目构建。等待直到您看到容器构建完成,然后转到下一节以测试您的新软件包。

  4. 运行以下命令以启动您的容器环境。

    1nvwb start --container
    

通过在代码中使用新软件包进行测试#

在您将软件包添加到您的项目并重建容器环境后,您可以在您的代码中使用新的软件包。

  1. 要打开 JupyterLab,请运行以下命令。

    JupyterLab 在您的默认浏览器中打开,您可以开始工作了。

    1nvwb start jupyterlab
    
  2. 在 JupyterLab 中,导航到 code 文件夹。如果您的项目已经有一个 notebook,请单击以打开它。否则,通过选择 Python 3 (ipykernel) 创建一个新的 notebook。

  3. 在您的 notebook 中,输入以下代码,然后单击运行。

    import numpy as np
    
  4. 输入以下代码,然后单击运行。

    print(np.e)
    

    您应该看到以下输出,这是 e 的值。

    2.718281828459045
    
  5. 保存您的 notebook。

  6. 在 JupyterLab 中,选择文件,然后单击关闭。确认关闭并关闭浏览器选项卡。

  7. 在您的命令 shell 中,运行以下命令以停止您的项目环境。

    1nvwb stop --container
    

下一步#