Tensorflow 动态显存

雾暗篷

于 2020-05-20 11:33:33 发布

阅读量433

点赞数 1

分类专栏： Tensorflow python 文章标签： tensorflow python

版权

7 篇文章

订阅专栏

3 篇文章

订阅专栏

本文介绍如何在TensorFlow中配置会话以优化GPU资源使用，通过设置允许GPU资源增长，避免一次性占用所有GPU内存，实现更高效的资源利用。

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

''' Start Session '''
config=tf.ConfigProto()
config.gpu_options.allow_growth=True
sess = tf.Session(config=config)