I am just trying to use my gpu with tensorflow on a Ubuntu 16.04 64bit system. I noticed while running some tensorflow commands that neither my cpu nor my gpu were using all of the configurations available. Thus, I updated the driver for my Nvidia GTX-970 card to x86_64-384.98, updated the toolkit to 9.0, and the cuDNN to 7.0.
I have verified the installation and can write some simple code using python to wrap Cuda language to access the GPU …