Megosztás:


GPU-specifikus csomagok beállítása Windows

Ez a szakasz ismerteti azokat a csomagokat, amelyekre az NVIDIA GPU-k használatához CNTK telepítenie kell.

A GPU-kompatibilitás ellenőrzése

A CNTK GPU-képességek használatához CUDA-kompatibilis grafikus kártyára van szükség. Ellenőrizheti, hogy a kártya CUDA-kompatibilis-e itt és itt (a régebbi kártyák esetében). A GPU-kártya számítási kapacitásának (CC) 3.0-s vagy újabbnak kell lennie.

A következő lépésekben telepíteni fogjuk a Microsoft Cognitive Toolkit létrehozásához szükséges NVidia fejlesztői eszközöket, valamint az NVidia támogatási kódtárait. Utolsó lépésként (miután telepítette az összes fent említett NVidia-eszközt!), ellenőriznie kell, hogy telepítve van-e a legújabb grafikuskártya-illesztőprogram.

Győződjön meg arról, hogy a címtár C:\Program Files\NVIDIA Corporation\NVSMI létezik a rendszerben.

  • Gyors telepítés ellenőrzése: Ha követte a fenti utasításokat, és ugyanazokat az elérési utakat használta, a parancs dir C:\Program Files\NVIDIA Corporation\NVSMI\nvml.dll sikeres lesz.

A legújabb GPU-kártyaillesztő

Telepítse a GPU-kártya legújabb illesztőprogramját:

  • Válassza ki a kártyát, és töltse le az illesztőprogram-csomagot erről a letöltési helyről
  • Az illesztőprogram telepítési eljárásának futtatása

NVIDIA CUDA 9.0

Töltse le és telepítse az NVIDIA CUDA 9.0 eszközkészletet:

Győződjön meg arról, hogy a következő CUDA környezeti változók a megfelelő elérési útra vannak beállítva (az NVIDIA Cuda telepítője hozza létre ezeket az Ön számára). A rendszer az alapértelmezett telepítési útvonalakat feltételezi:

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"
  • Gyors telepítés ellenőrzése: Ha követte a fenti utasításokat, és ugyanazokat az elérési utakat használta, a parancs dir C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin\cudart64_90.dll sikeres lesz.

cuDNN

Telepítse NVIDIA CUDA Deep Neural Network library az NVIDIA: cuDNN 7.0-s verzióban a CUDA 9.0-hoz készült cuDNN-t ezen a linken. Ez a verzió Windows 8.1-es, Windows 10- és Windows Server 2012 R2-es és újabb verziókhoz használható.

  • Bontsa ki az archívumot a helyi lemez egyik mappájába, például: C:\local\cudnn-9.0-v7.0\

  • Gyors telepítés ellenőrzése: Ha követte a fenti utasításokat, és ugyanazokat az elérési utakat használta, a parancs dir C:\local\cudnn-9.0-v7.0\cuda\bin\cudnn64_7.dll sikeres lesz.

KÖLYÖK

Fontos

Ha Python CNTK telepít, kihagyhatja ezt a lépést.

Fontos

Telepítse az NVIDIA CUB-t az alább megadott pontos verzióval. Erre azért van szükség, mert a CNTK buildkonfigurációs program ezt elvárja.

  • Töltse le az NVIDIA CUB v.1.7.4-et erről a letöltési hivatkozásról

  • Bontsa ki az archívumot a helyi lemez egyik mappájába (feltételezzük c:\local\cub-1.7.4).

  • Gyors telepítés ellenőrzése. Ha követte a fenti utasításokat, és ugyanazokat az elérési utakat használta, ez a parancs dir C:\local\cub-1.7.4\cub\cub.cuh sikeres lesz.