如何找到我的windows系统中安装了哪个TensorFlow？无论是CPU还是GPU TensorFlow

Question

我在Anaconda命令提示符下尝试了以下命令，

pip show tensorflow 

这给了我结果，

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: opensource@google.com
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

我想知道安装的TensorFlow是支持CPU还是支持GPU。我怎样才能得到这些信息？

Answer 1

TensorFlow pip包包括对启用CUDA卡的GPU支持

pip show tensorflow

对于较早版本的TensorFlow：

对于1.15及更高版本，CPU和GPU包是分开的：

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

因此，包名在1.15版和更高版本中是不同的。