Blog‎ > ‎

Installing TensorFlow on 10 with GPU

posted Feb 3, 2018, 8:12 PM by MUHAMMAD MUN`IM AHMAD ZABIDI   [ updated Feb 4, 2018, 12:26 AM ]
This post supersedes my earlier post on simply installing CUDA.

This is my version, for my Windows 10 machine, with GPU.

  • Install driver R390:

http://us.download.nvidia.com/Windows/390.77/390.77-desktop-win10-64bit-international-whql.exe

  • Install CUDA Toolkit 8.0:
  • Double click on the installer until you get this dialog:


  • After installation, go to this directory to check existence of sample programs
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\bin\win64\Release

  • cuDNN v6.0
To cuDNN, joing the NVIDiA developer network at https://developer.nvidia.com/cudnn. Then choose cuDNN v6.0 from the list

The Tensorflow documentation says "the cuDNN version must match exactly: TensorFlow will not load if it cannot find cuDNN64_6.dll."

References:
  • http://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html
  • https://www.tensorflow.org/install/install_windows#requirements_to_run_tensorflow_with_gpu_support
  • http://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html
Comments