我在 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。我如何获得此信息?
答案 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 及更早版本的包名称不同。