Tensorflow not using the GPU

Question

I have installed tensorflow-gpu using pip (within Anaconda). Here is the verification:

$ pip list | grep tensorflow tensorflow-gpu (1.4.0)

But whenever I run a model, it uses the CPU. Can someone please help.

Another command I tried:

$ find . -name tensorflow ./anaconda3/lib/python3.6/site-packages/tensorflow ./anaconda3/lib/python3.6/site-packages/tensorflow/include/tensorflow

Is there a conflict?

Faced the similar issue. My situation was tensorflow-gpu was installed using conda, while pip installed tensorflow cpu version. I re-installed tensorflow-gpu using pip fixed the issue. — Mingjiang Shi
– Mingjiang Shi, Commented May 29, 2018 at 11:49

Abhijit Balaji · Accepted Answer · 2018-02-03 04:14:56Z

4

Assuming you are using a Nvidia-gpu have you installed cuda and cudnn before installing Tensorflow with gpu support? check this link. Moreover use pip or pip3 to install tensorflow because Anaconda will not have the latest version of tensorflow.

also try running the following in a python or a ipython shell

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

You will get a sample output and if you see your GPU device in the message then tensorflow is using your gpu

edited Feb 3, 2018 at 4:14

answered Feb 3, 2018 at 4:07

Abhijit Balaji

1,9604 gold badges20 silver badges40 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Lakshmi Narayanan Over a year ago

I tried the same, and I can see my gpu in the list. I have an MSI. using my Dragon centre, which shows the stats and metrics of my hardware in function, I can see that my GPU is not being used, but rather, all of my RAM is being used. Is it something that cannot be picked by the metrics? Also, when I am running my games, I can see that the GPU is being used.

Abhijit Balaji Over a year ago

@LakshmiNarayanan Do you have a nvidia GPU or a AMD?

Collectives™ on Stack Overflow

Tensorflow not using the GPU

1 Answer 1

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related