Tensorflow on the GTX 1050

I have a Windows 10 laptop with a GEFORCE GTX 1050. On the NVIDIA developers site it is NOT listed as being supported by the CUDA tools, and therefore, I first assumed, I would not be ble to run GPU supported tensorflow on this machine. However that was a mistake: it is possible and the speed improvement is impressive.

  1. Download the CUDA 8.0 toolkit from https://developer.nvidia.com/cuda-downloads
  2. Install it, without the drivers, ignore the error message that says it does not see a supported GPU
  3. Also download and install the patch from the above website.
  4. Create a (free) developers account on the developers site.
  5. Download cudNN 6.0 after logging in. Also download the installtion guide.
  6. Copy the cudNN files to the location C:\Program Files\NVIDIA GPU
    Computing Toolkit\CUDA\v8.0
  7. Check whether the Path for CUDA has been set (see the installation guide if you do not know how to do that).

That’s it. Very simple but it took me hours to figure out.

In R, now you can download the keras package and issues the commands:


and you are ready to rock.

In the first test I ran it was 213 second without gpu and 18 sec with GPU sopprt.

This entry was posted in Uncategorized. Bookmark the permalink.

Leave a Reply

Fill in your details below or click an icon to log in:

WordPress.com Logo

You are commenting using your WordPress.com account. Log Out /  Change )

Google photo

You are commenting using your Google account. Log Out /  Change )

Twitter picture

You are commenting using your Twitter account. Log Out /  Change )

Facebook photo

You are commenting using your Facebook account. Log Out /  Change )

Connecting to %s