2017年4月6日
tensorflow-gpu 使用的常见错误
摘要： 这篇博客会不定期整理我在 tensorflow 中出现的问题和坑。 1. CUDA_ERROR_OUT_OF_MEMORY： tensorflow 在执行过程中会默认使用全部的 GPU 内存，给系统保留 200 M，但是在我的系统上会在分配内存时被拒绝导致报错，因此我们可以使用如下语句指定 GPU 阅读全文
posted @ 2017-04-06 19:42 仰望高端玩家的小清新 阅读(19176) 评论(0) 推荐(1) 编辑