Nastavení balíčků specifických pro GPU na Windows

Tato část popisuje balíčky, které potřebujete nastavit, aby CNTK využívat grafické procesory NVIDIA.

Kontrola kompatibility GPU

K používání CNTK funkcí GPU potřebujete grafickou kartu kompatibilní s CUDA. Tady můžete zkontrolovat, jestli je vaše karta kompatibilní s CUDA a tady (pro starší karty). Výpočetní funkce karty GPU (CC) musí být 3.0 nebo více.

V následujících krocích nainstalujeme vývojové nástroje NVidia potřebné k sestavení Microsoft Cognitive Toolkit a také knihoven podpory NVidia. Jako poslední krok (po instalaci všech výše uvedených nástrojů NVidia!) byste měli zkontrolovat, zda máte nainstalovaný nejnovější ovladač grafické karty.

Ujistěte se, C:\Program Files\NVIDIA Corporation\NVSMI že adresář existuje ve vašem systému.

• Rychlá kontrola instalace: Pokud jste postupovali podle pokynů výše a použili stejné cesty, příkaz dir C:\Program Files\NVIDIA Corporation\NVSMI\nvml.dll bude úspěšný.

Nejnovější ovladač karty GPU

Nainstalujte nejnovější ovladač pro kartu GPU:

• Vyberte kartu a stáhněte balíček ovladačů z tohoto umístění pro stažení.
• Spuštění instalačního postupu ovladače

NVIDIA CUDA 9.0

Stáhněte a nainstalujte Toolkit NVIDIA CUDA 9.0:

• Stažení požadovaného balíčku z této stránky pro stažení
• Spuštění instalačního postupu

Ujistěte se, že jsou následující proměnné prostředí CUDA nastavené na správnou cestu (instalační program NVIDIA Cuda je vytvoří za vás). Předpokládá se výchozí instalační cesty:

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"

• Rychlá kontrola instalace: Pokud jste postupovali podle pokynů výše a použili stejné cesty, příkaz dir C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin\cudart64_90.dll bude úspěšný.

cuDNN

Nainstalujte NVIDIA CUDA Deep Neural Network library také cuDNN ve verzi NVIDIA: cuDNN v7.0 pro CUDA 9.0 z tohoto odkazu. Tato verze je vhodná pro Windows 8.1, Windows 10 a také pro Windows Server 2012 R2 a novější.

• Extrahujte archiv do složky na místním disku, např. C:\local\cudnn-9.0-v7.0\

• Rychlá kontrola instalace: Pokud jste postupovali podle pokynů výše a použili stejné cesty, příkaz dir C:\local\cudnn-9.0-v7.0\cuda\bin\cudnn64_7.dll bude úspěšný.

MLÁDĚ

Důležité

Pokud instalujete CNTK pro Python, můžete tento krok přeskočit.

Důležité

Nainstalujte NVIDIA CUB pomocí přesné verze uvedené níže. To je nezbytné, protože je očekáváno CNTK konfiguračním programem sestavení.

• Stáhnout NVIDIA CUB v.1.7.4 z tohoto odkazu ke stažení

• Extrahujte archiv do složky na místním disku (předpokládáme c:\local\cub-1.7.4).

• Rychlá kontrola instalace Pokud jste postupovali podle výše uvedených pokynů a použili stejné cesty, tento příkaz dir C:\local\cub-1.7.4\cub\cub.cuh bude úspěšný.