笔记本是NVIDIA GeForce 940M的显卡,只有2G的显存,运行TensorFlow代码时候常出现OOM(Out of Memory)的错误,原因是batch_size设置得太大导致显存不足。如果想让代码仅仅运行在CPU下,可在原代码中加入如下代码:

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"  
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

   注:上述代码一定要放在import tensorflow或keras等之前,否则不起作用。

 

posted on 2018-03-19 21:24  艾克_塞伦特  阅读(10865)  评论(1编辑  收藏  举报