tensorflow2.1GPU显存限制无效的问题解决

xinxinbupp

于 2020-03-10 18:33:02 发布

阅读量1.6k

点赞数 2

分类专栏： tensorflow 文章标签： tensorflow gpu

本文链接：https://blog.csdn.net/qq_41280654/article/details/104780050

版权

本文介绍了在使用TensorFlow 2.1时遇到GPU显存限制设置无效的问题，以及如何通过API找到正确的解决方法。详细讨论了从TensorFlow v1到v2.1版本中显存限制的用法变化。

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

tensorflow2.1显存限制

查了一下午，查到的方法全都无法解决，在绝望的时候从API中找到了正解

opts = tf.GPUOptions(per_process_gpu_memory_fraction=0.5)
sess = tf.Session(config=tf.ConfigProto(gpu_options=opts