使用tensorflow2.x训练的时候关于GPU的设置

原创

于 2020-09-14 13:33:50 发布

· 1.8k 阅读

11 ·

版权

本文介绍了在TensorFlow2.x中如何查看和设置GPU使用，包括列出GPU列表、选择可见GPU、管理GPU内存以及单个GPU模拟多GPU训练的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.列出当前主机的物理显卡GPU或者CPU的列表

gpus = tf.config.experimental.list_physical_devices(device_type='GPU')
cpus = tf.config.experimental.list_physical_devices(device_type='CPU')
print(gpus)
打印结果
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]

2.设置当前程序可见GPU设备的方法两种方法

方法1


tf.config.experimental.set_visible_devices(devices

最低0.47元/天解锁文章