TensorFlow训练时合理安排GPU利用率

code • 2022-4-9 • python • 阅读 55

gpus = tf.config.experimental.list_physical_devices(device_type='GPU')
for gpu in gpus:
    tf.config.experimental.set_memory_growth(gpu, True)

将其插入训练文件中，例如下图

欢迎分享，转载请注明来源：内存溢出

打赏

微信扫一扫

支付宝扫一扫

上一篇 2022-04-09

下一篇 2022-04-09

登录后才能评论