如何找到Windows系统中安装了哪个TensorFlow?无论是CPU还是GPU TensorFlow

时间:2018-09-01 06:34:16

标签: python tensorflow anaconda prompt

我在 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。我如何获得此信息?

1 个答案:

答案 0 :(得分: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 及更早版本的包名称不同。