Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Tensorflow doesn't seem to see my gpu

Tags:

tensorflow

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

like image 512
Nikita Petrenko Avatar asked Dec 30 '16 20:12

Nikita Petrenko


People also ask

Why is TensorFlow not detecting my GPU?

This is most likely because the CUDA and CuDNN drivers are not being correctly detected in your system. In both cases, Tensorflow is not detecting your Nvidia GPU. This can be for a variety of reasons: Nvidia Driver not installed.

Does TensorFlow automatically detect GPU?

If a TensorFlow operation has both CPU and GPU implementations, TensorFlow will automatically place the operation to run on a GPU device first. If you have more than one GPU, the GPU with the lowest ID will be selected by default. However, TensorFlow does not place operations into multiple GPUs automatically.


1 Answers

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

like image 80
Kenan Avatar answered Oct 05 '22 13:10

Kenan