如何在模型產生器中安裝 GPU 支援

瞭解如何安裝 GPU 驅動程式，以搭配模型產生器使用 GPU。

硬體需求

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

先決條件

• 模型產生器 Visual Studio 延伸模組。 擴充功能內建於Visual Studio16.6.1版。
• 請確定已針對 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. 安裝最新的驅動程式。