Delen via

GPU-specifieke pakketten instellen op Windows

  • Artikel

In deze sectie vindt u een overzicht van de pakketten die u moet instellen, zodat CNTK gebruik kan maken van NVIDIA GPU's.

Uw GPU-compatibiliteit controleren

U hebt een CUDA-compatibele grafische kaart nodig om CNTK GPU-mogelijkheden te gebruiken. U kunt controleren of uw kaart hier en hier (voor oudere kaarten) compatibel is met CUDA. Uw GPU-kaart Compute Capability (CC) moet 3.0 of meer zijn.

In de volgende stappen installeren we de NVidia-ontwikkelhulpprogramma's die nodig zijn om de Microsoft Cognitive Toolkit en NVidia-ondersteuningsbibliotheken te bouwen. Als laatste stap (nadat u alle genoemde NVidia-hulpprogramma's hebt geïnstalleerd!), moet u controleren of het meest recente stuurprogramma voor de grafische kaart is geïnstalleerd.

Zorg ervoor dat de map C:\Program Files\NVIDIA Corporation\NVSMI in uw systeem bestaat.

  • Snelle installatiecontrole: Als u de bovenstaande instructies hebt gevolgd en dezelfde paden hebt gebruikt, slaagt de opdracht dir C:\Program Files\NVIDIA Corporation\NVSMI\nvml.dll .

Meest recente GPU-kaartstuurprogramma

Installeer het meest recente stuurprogramma voor uw GPU-kaart:

  • Selecteer uw kaart en download het stuurprogrammapakket vanaf deze downloadlocatie
  • De installatieprocedure voor stuurprogramma's uitvoeren

NVIDIA CUDA 9.0

Download en installeer de NVIDIA CUDA 9.0 Toolkit:

Zorg ervoor dat de volgende CUDA-omgevingsvariabelen zijn ingesteld op het juiste pad (het NVIDIA Cuda-installatieprogramma maakt deze voor u). Standaardinstallatiepaden worden aangenomen:

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"
  • Snelle installatiecontrole: Als u de bovenstaande instructies hebt gevolgd en dezelfde paden hebt gebruikt, slaagt de opdracht dir C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin\cudart64_90.dll .

cuDNNN

Installeer NVIDIA CUDA Deep Neural Network library ook wel cuDNN genoemd in de versie NVIDIA: cuDNN v7.0 voor CUDA 9.0 via deze koppeling. Deze versie is geschikt voor Windows 8.1, Windows 10 en Windows Server 2012 R2 en hoger.

  • Pak het archief uit naar een map op uw lokale schijf, bijvoorbeeld naar C:\local\cudnn-9.0-v7.0\

  • Snelle installatiecontrole: Als u de bovenstaande instructies hebt gevolgd en dezelfde paden hebt gebruikt, slaagt de opdracht dir C:\local\cudnn-9.0-v7.0\cuda\bin\cudnn64_7.dll .

CUB

Belangrijk

Als u CNTK installeert voor Python, kunt u deze stap overslaan.

Belangrijk

Installeer NVIDIA CUB met de exacte versie die hieronder is opgegeven. Dit is nodig omdat dit wordt verwacht door het CNTK buildconfiguratieprogramma.

  • Download NVIDIA CUB v.1.7.4 via deze downloadkoppeling

  • Pak het archief uit naar een map op uw lokale schijf (we gaan ervan uit).c:\local\cub-1.7.4

  • Snelle installatiecontrole. Als u de bovenstaande instructies hebt gevolgd en dezelfde paden hebt gebruikt, slaagt deze opdracht dir C:\local\cub-1.7.4\cub\cub.cuh .