gpt4 book ai didi

tensorflow - 使用 tensorflow 后端时在 Keras 中使用 tensorflow.GPUOptions

转载 作者:行者123 更新时间:2023-12-04 17:31:54 25 4
gpt4 key购买 nike

在 tensorflow 中,我可以在创建 session 时执行以下操作:

tf.GPUOptions(per_process_gpu_memory_fraction=0.333,allow_growth=True)

有没有办法用 tensorflow 后端在 keras 中做同样的事情?

最佳答案

您可以使用 keras.backend.tensorflow_backend.set_session() 设置 Keras 全局 tensorflow session :

import tensorflow as tf
import keras.backend.tensorflow_backend as ktf


def get_session(gpu_fraction=0.333):
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=gpu_fraction,
allow_growth=True)
return tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))


ktf.set_session(get_session())

关于tensorflow - 使用 tensorflow 后端时在 Keras 中使用 tensorflow.GPUOptions,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41762272/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com