tensorflow显存不够使用如何解决(windows系统)

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

查看下显卡发现果然几乎用完了,具体查看步骤请参考:https://blog.csdn.net/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.csdn.net/leibaojiangjun1/article/details/53671257

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

https://blog.csdn.net/liangdong2014/article/details/78860661

https://blog.csdn.net/lgh0824/article/details/77096241

 

展开阅读全文

没有更多推荐了,返回首页