How to prevent tensorflow from allocating the totality of a GPU memory?

前端 未结 16 2215
南旧
南旧 2020-11-22 04:26

I work in an environment in which computational resources are shared, i.e., we have a few server machines equipped with a few Nvidia Titan X GPUs each.

For small to m

16条回答
  •  时光说笑
    2020-11-22 04:55

    # allocate 60% of GPU memory 
    from keras.backend.tensorflow_backend import set_session
    import tensorflow as tf 
    config = tf.ConfigProto()
    config.gpu_options.per_process_gpu_memory_fraction = 0.6
    set_session(tf.Session(config=config))
    

提交回复
热议问题