共用方式為

在 Windows 上設定 GPU 特定套件

  • 發行項

本節概述您需要設定的套件，以便CNTK利用 NVIDIA GPU。

檢查您的 GPU 相容性

您需要與 CUDA相容的圖形卡，才能使用 CNTK GPU 功能。 您可以 在這裡 檢查卡片是否與 CUDA 相容， 在這裡 (舊版卡片) 。 您的 GPU 記憶卡 計算功能 (CC) 必須為 3.0 或以上。

在下列步驟中，我們將安裝建置Microsoft Cognitive Toolkit以及 NVidia 支援程式庫所需的 NVidia 開發工具。 在您安裝上述所有 NVidia tools！) 之後，最後一個步驟 (，您應該檢查是否已安裝最新的圖形卡驅動程式。

請確定您的系統中存在目錄 C:\Program Files\NVIDIA Corporation\NVSMI 。

  • 快速安裝檢查：如果您遵循上述指示並使用相同的路徑，命令 dir C:\Program Files\NVIDIA Corporation\NVSMI\nvml.dll 將會成功。

最新的 GPU 記憶卡驅動程式

安裝 GPU 記憶卡的最新驅動程式：

NVIDIA CUDA 9.0

下載並安裝 NVIDIA CUDA 9.0 工具組：

請確定下列 CUDA 環境變數設定為正確的路徑， (NVIDIA Cuda 安裝程式會為您建立這些) 。 預設安裝路徑假設為：

CUDA_PATH="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0"
CUDA_PATH_V9_0="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0"
  • 快速安裝檢查：如果您遵循上述指示並使用相同的路徑，命令 dir C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin\cudart64_90.dll 將會成功。

cuDNN

在此連結中安裝 NVIDIA CUDA Deep Neural Network library 也稱為cuDNN版本 NVIDIA：cuDNN v7.0 for CUDA 9.0。 此版本適用于Windows 8.1、Windows 10，以及 Windows Server 2012 R2 和更新版本。

  • 將封存解壓縮到本機磁片上的資料夾，例如 C:\local\cudnn-9.0-v7.0\

  • 快速安裝檢查：如果您遵循上述指示並使用相同的路徑，命令 dir C:\local\cudnn-9.0-v7.0\cuda\bin\cudnn64_7.dll 將會成功。

CUB

重要

如果您要為 Python 安裝CNTK，您可以略過此步驟。

重要

使用下面指定的確切版本來安裝 NVIDIA CUB。 這是必要的，因為它必須是CNTK組建組態程式。

  • 從此下載連結下載NVIDIA CUB v.1.7.4

  • (我們假設 c:\local\cub-1.7.4) ，將封存解壓縮到本機磁片上的資料夾。

  • 快速安裝檢查。 如果您遵循上述指示並使用相同的路徑，此命令 dir C:\local\cub-1.7.4\cub\cub.cuh 將會成功。