Tensorflow设置gpu占有率

最新推荐文章于 2021-08-16 20:48:55 发布

cool_策

最新推荐文章于 2021-08-16 20:48:55 发布

阅读量757

点赞数

分类专栏： tensorflow 深度学习

本文链接：https://blog.csdn.net/qq_34510308/article/details/105066421

版权

深度学习同时被 2 个专栏收录

25 篇文章 1 订阅

订阅专栏

tensorflow

8 篇文章 0 订阅

订阅专栏

调整gpu_fraction参数大小（最大设置为1.0）

def get_session(gpu_fraction=0.5):

    num_threads = os.environ.get('OMP_NUM_THREADS')
    gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=gpu_fraction)

    if num_threads:
        return tf.Session(config=tf.ConfigProto(
            gpu_options=gpu_options, intra_op_parallelism_threads=num_threads))
    else:
        return tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))