Tensorflow设置GPU训练模型

Posted 2021-12-29 ZSYL

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了Tensorflow设置GPU训练模型相关的知识，希望对你有一定的参考价值。

添加以下代码

import tensorflow as tf

gpus = tf.config.list_physical_devices("GPU")
if gpus:
    tf.config.experimental.set_memory_growth(gpus[0],True)

加油!

感谢!

努力!

以上是关于Tensorflow设置GPU训练模型的主要内容，如果未能解决你的问题，请参考以下文章