跑深度学习时tensorflow-gpu内存不够使用如何解决
跑深度学习时tensorflow-gpu不够使用如何解决今天跑一个深度学习代码时遇到一个问题:Hint: If you want to see a list of allocated tensors when OOM happen百度了之后发现是gpu内存不够使用两个解决方案1.制定第二块GPU可用import osos.environ["CUDA_VISIBLE_DEVICES"] = "2"2.调整batch_size的大小400张训练集,我的batch_size原本为8,跑50次出
原创
2020-11-21 19:38:26 ·
5982 阅读 ·
4 评论