筆記本是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等之前，否則不起作用。

?

轉載于:https://www.cnblogs.com/excellent-ship/p/8604941.html

總結

以上是生活随笔為你收集整理的限制TensorFlow只在CPU上运行的方法的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。