2018 年 3月 19 日随笔档案 - 艾克_塞伦特

2018年3月19日

摘要：笔记本是NVIDIA GeForce 940M的显卡，只有2G的显存，运行TensorFlow代码时候常出现OOM(Out of Memory)的错误，原因是batch_size设置得太大导致显存不足。如果想让代码仅仅运行在CPU下，可在原代码中加入如下代码：注：上述代码一定要放在import t 阅读全文

posted @ 2018-03-19 21:24 艾克_塞伦特阅读(10888) 评论(1) 推荐(2) 编辑