筆記本是NVIDIA GeForce 940M的顯卡，只有2G的顯存，運(yùn)行TensorFlow代碼時(shí)候常出現(xiàn)OOM(Out of Memory)的錯(cuò)誤，原因是batch_size設(shè)置得太大導(dǎo)致顯存不足。如果想讓代碼僅僅運(yùn)行在CPU下，可在原代碼中加入如下代碼：

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

?　　注：上述代碼一定要放在import tensorflow或keras等之前，否則不起作用。

?

轉(zhuǎn)載于:https://www.cnblogs.com/excellent-ship/p/8604941.html

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。