如何在 Model Builder 中安裝 GPU 支援

了解如何安裝 GPU 驅動程式，以搭配 Model Builder 使用 GPU。

硬體需求

• 至少一個 CUDA 相容 GPU。 如需相容的 GPU 清單，請參閱 NVIDIA 指南。
• 至少 6GB 的專用 GPU 記憶體。

必要條件

• Model Builder Visual Studio 延伸模組。 延伸模組自 Visual Studio 16.6.1 版起內建於其中。
• 請確定已針對 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 類型？

1. 以滑鼠右鍵按一下桌面
2. 如果您在快顯視窗中看到 [NVIDIA 控制面板] 或 [NVIDIA 顯示器]，則表示您有 NVIDIA GPU
3. 按一下快顯視窗中的 [NVIDIA 控制面板] 或 [NVIDIA 顯示器]
4. 查看 [圖形卡資訊]
5. 您將會看到 NVIDIA GPU 的名稱

我看不到 NVIDIA 控制面板 (或無法開啟)，但我知道我有 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. 安裝最新的驅動程式。