tensorflow显存不够使用如何解决（windows系统）

本人在跑完训练部分，然后开始跑测试部分出现了以下问题，这个问题困扰了一上午

查看下显卡发现果然几乎用完了，具体查看步骤请参考：https://blog.youkuaiyun.com/zhuimengshaonian66/article/details/81286206

其实之前也出现过一次这种问题，忘记怎么解决了，好像是重启，不建议大家去试，因为下面有完美解决这个问题的办法

我是在 main函数里面加入这几句话就解决了这个问题

为了便于各位复制粘贴

import tensorflow as tf
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0'   #指定第一块GPU可用
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.5  # 程序最多只能占用指定gpu50%的显存
config.gpu_options.allow_growth = True      #程序按需申请内存
sess = tf.Session(config = config)

然后就跑通测试部分了，很神奇。

以上参考：https://blog.youkuaiyun.com/leibaojiangjun1/article/details/53671257

还有一种办法，不过是linux系统的，各位有兴趣可以试下：

https://blog.youkuaiyun.com/liangdong2014/article/details/78860661

https://blog.youkuaiyun.com/lgh0824/article/details/77096241