tensorflow显存不够使用如何解决（windows系统） |
您所在的位置：网站首页 › 显存使用量超过显存可用量该怎么解决 › tensorflow显存不够使用如何解决（windows系统） |
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
|
【本文地址】
今日新闻 |
推荐新闻 |