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

Question

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

我不想运行一个微不足道的模型（与log_device_placement）必须弄清楚。

我试过使用tensorflow.__version__，但好像1.3.0-rc2是双面打印的。

Answer 1

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

如果你想以编程内蟒蛇检查，这是一条路可走：

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 2

是它通过PIP安装？您可以检查pip list，它会告诉之一：

tensorflow-gpu 

或

tensorflow 

二是CPU版本的答案