Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How to find which TensorFlow is installed in my windows system? Whether it is CPU or GPU TensorFlow

I tried out the following command on the Anaconda command prompt,

pip show tensorflow 

which gave me the result,

Name: tensorflow
Version: 1.10.0
Summary: TensorFlow is an open source machine learning framework for everyone.
Home-page: https://www.tensorflow.org/
Author: Google Inc.
Author-email: [email protected]
License: Apache 2.0
Location: d:\softwares\pycond\envs\tensorflow2\lib\site-packages
Requires: termcolor, absl-py, grpcio, tensorboard, wheel, protobuf, gast, six, numpy, setuptools, astor

I want to know if the installed TensorFlow is CPU supported or GPU supported. How can I get this information?

like image 618
Rahul Meena Avatar asked Sep 01 '18 06:09

Rahul Meena


People also ask

How do I know which GPU is being used by TensorFlow?

tf. test. gpu_device_name returns the name of the gpu device.

What is TensorFlow CPU and GPU?

TensorFlow refers to the CPU on your local machine as /device:CPU:0 and to the first GPU as /GPU:0—additional GPUs will have sequential numbering. By default, if a GPU is available, TensorFlow will use it for all operations.


1 Answers

The TensorFlow pip package includes GPU support for CUDA®-enabled cards

pip show tensorflow

For Older versions of TensorFlow:

For releases 1.15 and older, CPU and GPU packages are separate:

pip install tensorflow==1.15      # CPU
pip install tensorflow-gpu==1.15  # GPU

So, package names are different in for releases 1.15 and older.

like image 94
Rohith Avatar answered Oct 02 '22 22:10

Rohith