Tensorflow doesn't seem to see my gpu

Question

I've tried tensorflow on both cuda 7.5 and 8.0, w/o cudnn (my GPU is old, cudnn doesn't support it).

When I execute device_lib.list_local_devices(), there is no gpu in the output. Theano sees my gpu, and works fine with it, and examples in /usr/share/cuda/samples work fine as well.

I installed tensorflow through pip install. Is my gpu too old for tf to support it? gtx 460

Answer 1

I came across this same issue in jupyter notebooks. This could be an easy fix.

$ pip uninstall tensorflow
$ pip install tensorflow-gpu

You can check if it worked with:

tf.test.gpu_device_name()

Update 2020

It seems like tensorflow 2.0+ comes with gpu capabilities therefore pip install tensorflow should be enough