Setting Up Tensorflow (with CUDA) for Windows 10



Below are some of my rough notes on how I've setup my Windows 10 laptop to use Tensorflow with CUDA.

My reference:
  1. http://www.heatonresearch.com/2017/01/01/tensorflow-windows-gpu.html
  2. https://www.tensorflow.org/install/
Install the following NVidia drivers:
  1. CUDA Drivers (http://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows). Currently it's CUDA Toolkit 8.0. Can install using the installer downloaded.
  2. CUDNN - CUDA for Deep Neural Networks (https://developer.nvidia.com/cudnn). Currently it's 5.1. Once extracted, place the files in the respective directory along with the other CUDA files in the NVIDIA Toolkit folder
Setting up Tensorflow (CPU)

Setting up Tensorflow (GPU)

Note: It's 22/2/2017 now and Google have recently released their Tensorflow 1.0, which might've rendered the above guide obsolote (i've haven't tested them yet).

Update (23/2/2017):
The above basically creates 2 new anaconda instance for you to play with. I like this approach since it's cleaner and does not mess up my original Anaconda installation - easier to troubleshoot (and destroy if need be). To select the instance that you wish to run, and run jupyter - you may use the following steps:

And your Jupyter should've already loaded by now.

Comments

Popular posts from this blog

HIVE: Both Left and Right Aliases Encountered in Join

Assign select result to variable in Netezza stored procedure

Splitting value in Netezza using array_split