Reputation: 79
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?
Upvotes: 7
Views: 16162
Reputation: 246
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.
Upvotes: 1