如何判断我是否使用python安装了tensorflow-gpu？

Question

我试图在python中区分用户是否已安装tensorflow-gpu或仅tensorflow（在CPU上，缺少GPU支持）。

我不想运行一个简单的模型（使用log_device_placement）来解决这个问题。

我尝试过使用tensorflow.__version__，但似乎1.3.0-rc2双面打印。

Answer 1

是否通过pip安装？您可以查看pip list，它会显示：

tensorflow-gpu

或

tensorflow

第二个是cpu版本

Answer 2

运行pip freeze | grep tensorflow如果已安装，您会在结果中看到tensorflow-gpu。

如果你想在python中以编程方式检查它，这是一种方法：

import pip 

l = next(str(i) for i in pip.get_installed_distributions() if 'tensorflow-gpu' in str(i))
print(l)

在我的情况下输出：

tensorflow-gpu 0.12.0rc0

Answer 3

Mohamed的答案在Python 3.6及更高版本中不再起作用，但以下内容可以起作用：

import pkg_resources
l = [d for d in pkg_resources.working_set  if 'tensorflow' in str(d)]
print(l)