使用tensorflow时，关于GPU的设置

Posted 2021-12-24 hellojiaojiao

tags:

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

查看显卡使用情况：

设置tensorflow按需分配资源：

import os

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"

os.environ["CUDA_VISIBLE_DEVICES"] = "1" # 使用第二块GPU（从0开始）

gpu_options = tf.GPUOptions(allow_growth=True)

sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

以上是关于使用tensorflow时，关于GPU的设置的主要内容，如果未能解决你的问题，请参考以下文章