-
Notifications
You must be signed in to change notification settings - Fork 4.3k
CNTK move to Cuda8
With the release of CNTK 2 Beta 5 (November 24th, 2016) the toolkit moved to NVIDIA CUDA 8 as the development toolkit for GPU accelerated applications. This allows the Microsoft Cognitive Toolkit to take advantage of the latest improvements in NVIDIA graphic cards and optimizations.
As the first step this move is implemented on the Windows platform. The support of CUDA 8 for the Microsoft Cognitive Toolkit on Linux will follow soon!
Currently this is an optional move. That means you still can do your own CNTK development and testing with NVIDIA CUDA 7.5, but we at Microsoft have moved our main Windows development environment and our Windows testing environment to CUDA 8. We will not do any further development or testing on CUDA 7.5. In the next few weeks we will make NVIDIA CUDA 8 a mandatory requirement for the Microsoft Cognitive Toolkit on Windows and Linux.
The CNTK binary packages you download starting with Beta 5 contains the NVIDIA CUDA 8 runtime components. You don't need to download any additional components, although we recommend you check the NVIDIA website for the latest set of drivers for your NVIDIA graphics card.
Setting up a development environment for the Microsoft Cognitive Toolkit from scratch is described here. If you have already a working development environment for CNTK you just need to follow these steps:
The NVIDIA CUDA 7.5 development environment and the CUDA 8.0 development environment can live side-by-side on your machine, but you can also deinstall the CUDA 7.5 toolkit.
Download and install the NVIDIA CUDA 8.0 Toolkit:
- Download the required package from this download page
- Run the installation procedure
Make sure that the following CUDA environment variables are set to the correct path (the installer will create these for you). Default installation paths are assumed:
CUDA_PATH="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0"
CUDA_PATH_V8_0="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0"
Install NVIDIA CUDA Deep Neural Network library also known as cuDNN in the verion NVIDIA:cuDNN v5.1 for CUDA 8.0 from this link. This version is suitable for Windows 8.1, Windows 10, as well as Windows Server 2012 R2.
- Extract the archive to a folder on your local disk, e.g. to
C:\local\cudnn-8.0-v5.1\and set the environment variableCUDNN_PATHpointing to that location, e.g.:
setx CUDNN_PATH C:\NVIDIA\cudnn-5.1\CUDA
- Quick installation check: If you followed the instruction above and used the same paths, the command
dir C:\local\cudnn-8.0-v5.1\CUDA\include\cudnn.hwill succeed.
Now install the latest driver for your GPU card:
- Select your card and download the driver pack from this download location
You can now restart Visual Studio and open the CNTK project. Please rebuild to make sure all components are recreated in the new environment.