| 跑深度学习时tensorflow | 您所在的位置：网站首页 › 电脑显存设置为不可用怎么办 › 跑深度学习时tensorflow |
跑深度学习时tensorflow
|
今天跑一个深度学习代码时遇到一个问题：Hint: If you want to see a list of allocated tensors when OOM happen
百度了之后发现是gpu内存不够使用 两个解决方案 1.制定第二块GPU可用 import os os.environ["CUDA_VISIBLE_DEVICES"] = "2" 2.调整batch_size的大小400张训练集，我的batch_size原本为8，跑50次出现内存爆炸的情况 调为4之后，100次，成功解决 BATCH_SIZE = 4
|
还有一些其他的方法，例如设置GPU的使用率等，博主没有实践，可参考：深度学习笔记：windows+tensorflow 指定GPU占用内存（解决gpu爆炸问题）【本文地址】
公司简介
联系我们