本文共 247 字，大约阅读时间需要 1 分钟。

import tensorflow as tfgpus = tf.config.list_physical_devices("GPU")print(gpus)if gpus:    gpu0 = gpus[0] #如果有多个GPU，仅使用第0个GPU    tf.config.experimental.set_memory_growth(gpu0, True) #设置GPU显存用量按需使用        tf.config.set_visible_devices([gpu0],"GPU")

转载地址：https://mtyjkh.blog.csdn.net/article/details/114886846 如侵犯您的版权，请留言回复原文章的地址，我们会给您删除此文章，给您带来不便请您谅解！