사용 tensorflow.GPUOptions tensorflow 백엔드

Question

를 사용하여 :

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

tensorflow 백엔드 keras에서 동일한 작업을 수행 할 수있는 방법이 있나요?

Answer 1

당신은 keras.backend.tensorflow_backend.set_session()와 Keras 글로벌 tensorflow 세션을 설정할 수 있습니다

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()) 

을