我开始使用 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 导入的。
