Saturday, March 11, 2017

install TensorFlow in Windows

1) install python3.5 https://www.python.org/downloads/
2) install Matplotlib `pip install matplotlib`
3) install tensorflow `pip install tensorflow`
Optional
4) install tensorflow_gpu `pip install tensorflow_gpu`
5) install CUDA8.0 https://developer.nvidia.com/cuda-downloads
6) download cuDNN https://developer.nvidia.com/cudnn (require membership with free registration), unzip it, and copy the files into related folder in CUDA [1]


`cuda\bin\cudnn64_5.dll` to `C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin\`
`cuda\include\cudnn.h` to `C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include\`
`cuda\lib\x64\cudnn.lib` to `C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64\`
7) Add CUDA path (C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin) into %PATH% environment variable

References:
[1] https://github.com/tensorflow/tensorflow/issues/5968

No comments:

Post a Comment