- 问题:tensorflow调gpu显存分配的问题。tensorflow默认申请可使用的全部显存,当tensorflow程序运行会话却没有关闭会话释放资源的时候,就会出现此错误。
- 解决办法:将建立会话后默认的分配全部现存改成按需分配。
- 具体:
在运行Session时将:
with tf.Session() as sess:
替换为
gpu_options = tf.GPUOptions(allow_growth=True)
with tf.Session(config
with tf.Session() as sess:
替换为
gpu_options = tf.GPUOptions(allow_growth=True)
with tf.Session(config