Skip to main content
 首页 » 编程设计

python之我怎么知道我是在导入 tensorflow 还是 tensorflow-gpu

2024年12月31日30artech

我开始使用 tensorflow cpu 版本,但后来我安装了 tensorflow-gpu 以加快处理速度,但我不确定当前使用的是哪个版本。如何确认当前运行的版本确实是tensorflow-gpu?

请您参考如下方法:

据我所知,您无法直接从 python 中判断您是从 tensorflow(仅限 CPU)还是 tensorflow-gpu 包导入 tensorflow。

不过,你可以向tensorflow询问它知道的设备:

from tensorflow.python.client.device_lib import list_local_devices 
print(list_local_devices()) 

如果有任何设备的 device_type 类型为 GPU,那么您肯定是从 tensorflow-gpu 导入的。