首页 > 解决方案 > 在 keras 中训练机器学习模型的 GPU 和 XLA_CPU 有什么区别?

问题描述

我想使用比 CPU 更好的处理器在 keras 中训练 ML 模型

如果我运行:

physical_devices = tf.config.list_physical_devices()
print(physical_devices)
# [
#   PhysicalDevice(name='/physical_device:CPU:0', device_type='CPU'),
#   PhysicalDevice(name='/physical_device:XLA_CPU:0', device_type='XLA_CPU')
# ]

标签: tensorflowgputensorflow2.0keras-2

解决方案


推荐阅读