如何在 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 類型?
從設定檢查 GPU
- 以滑鼠右鍵按一下 Windows [開始] 功能表圖示,然後選取 [設定]。
- 選取 [設定]> [系統]
- 選取 [顯示],然後向下捲動至 [相關設定]。
- 選取 [進階顯示]。 GPU 的品牌和型號應該會顯示在 [顯示資訊] 底下。
透過工作管理員檢查 GPU
- 以滑鼠右鍵按一下 Windows [開始] 功能表圖示,然後選取 [工作管理員]。
- 選取 [效能]。
- 在索引標籤的最後一個窗格中,選擇 [GPU]。 如果這是可用的選項,則其很可能位於清單底部。
- 在 GPU 選取項目的右上角會顯示電腦 GPU 的相關資訊。
我在 [設定] 或 [工作管理器] 中看不到 GPU,但知道我使用的是 NVIDIA GPU。
- 開啟 [裝置管理員]
- 查看顯示卡
- 為您的 GPU 安裝適當的驅動程式。
如何查看我所擁有的 CUDA 版本?
- 開啟 PowerShell 或命令列視窗
- 輸入
nvcc --version
無法使用 cuda,請確認您有支援 cuda 的 GPU
- 開啟 GeForce Experience 應用程式。
- 應用程式應該會顯示已安裝且可用的驅動程式更新。 如果您在看到更新時遇到問題,您可以從 https://www.nvidia.com/geforce/drivers/ 取得最新的驅動程式。
- 安裝最新的驅動程式。