如何查找我的 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\xc2\xae 的卡的 GPU 支持

\n

pip show tensorflow\n

\n

对于旧版本的 TensorFlow：

\n

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

\n

pip install tensorflow==1.15      # CPU\npip install tensorflow-gpu==1.15  # GPU\n

\n

因此，1.15 及更早版本中的包名称有所不同。

\n