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

我试图在python中区分用户是安装了

tensorflow gpu

，还是仅仅安装了

tensorflow

（在缺乏gpu支持的CPU上）

我不想运行一个简单的模型（使用

log\u device\u placement

）来解决这个问题

---

我试过使用tensorflow。uuu版本uuu，但似乎

1.3.0-rc2

是双向打印的。

运行

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

---

它是通过pip安装的吗？您可以检查

pip列表

，它将显示：

tensorflow-gpu

或

---

第二个问题是，Mohamed的cpu版本在Python 3.6+中不再有效，但下面的答案确实有效：

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

---

下面的答案都不是你需要的？

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