如何在 Model Builder 中安装 GPU 支持

了解如何安装 GPU 驱动程序，以便将 GPU 与 Model Builder 一起使用。

硬件要求

• 至少一个 CUDA 兼容 GPU。 有关兼容 GPU 的列表，请参阅 NVIDIA 指南。
• 至少 6 GB 专用 GPU 内存。

先决条件

• Model Builder Visual Studio 扩展。 自版本 16.6.1 起，该扩展内置于 Visual Studio 中。
• 请确保为 GPU 安装相应的驱动程序。

仅限图像分类

• NVIDIA 开发人员帐户。 如果没有帐户，请创建一个免费帐户。
• 安装依赖项
  • 安装 CUDA v10.1。 请确保安装 CUDA v10.1，而不是任何其他更新版本。
  • 安装 cuDNN v7.6.4 for CUDA 10.1。 不能安装多个版本的 cuDNN。 下载 cuDNN v7.6.4 zip 文件并将其解压缩后，将 <CUDNN_zip_files_path>\cuda\bin\cudnn64_7.dll 复制到 <YOUR_DRIVE>\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin。

疑难解答

如果未在本地安装 GPU，该怎么办？

深度学习方案在 GPU 上运行时往往速度更快。

某些方案（例如图像分类）支持在 Azure GPU VM 上进行训练。

但是，如果本地 GPU 或 Azure 不适合你，这些方案也会在 CPU 上运行。 请注意，这样的话，训练时间明显更长。

如何知道我拥有哪些 GPU？

通过“设置”检查 GPU

1. 右键单击 Windows 开始菜单图标，然后选择“设置”。
2. 选择“设置”>“系统”
3. 选择“显示器”，向下滚动到“相关设置”。
4. 选择“高级显示器”。 GPU 的品牌和型号应显示在“显示器信息”下。

通过“任务管理器”检查 GPU

1. 右键单击 Windows 开始菜单图标，然后选择“任务管理器”。
2. 选择“性能”。
3. 在选项卡的最后一个窗格中，选择 GPU。 如果此选项可用，则它可能位于列表底部。
4. GPU 选择的右上角将显示有关计算机的 GPU 信息。

我在“设置”或“任务管理器”中看不到我的 GPU，但我知道我有一个 NVIDIA GPU。

1. 打开“设备管理器”
2. 查看“显示适配器”
3. 为 GPU 安装适当的驱动程序 。

如何查看有哪些版本的 CUDA？

1. 打开 PowerShell 或命令行窗口
2. 键入 nvcc --version

CUDA 不可用，请确认你拥有 CUDA 支持的 GPU

1. 打开 GeForce Experience 应用。
2. 应用程序应会显示已安装和可用的驱动程序更新。 如果在查看更新时遇到问题，可以从 https://www.nvidia.com/geforce/drivers/ 获取最新的驱动程序。
3. 安装最新的驱动程序。