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

前端 未结 16 2205
南旧
南旧 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:49

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

    https://github.com/tensorflow/tensorflow/issues/1578

提交回复
热议问题