如何在 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 存档安装 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. 安装最新的驱动程序。