限制TensorFlow只在CPU上运行的方法

　　笔记本是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上运行的方法的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。