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

Posted 2021-01-17 yanshuai0922

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了限制TensorFlow只在CPU上运行的方法相关的知识，希望对你有一定的参考价值。

我们知道安装tensorflow可以安装两个版本的，有cpu和gpu的，如果同时安装的时候会默认进行用gpu进行，那如歌更改为使用cpu呢

在import tensorflow/keras 之前，加入如下代码

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

以上是关于限制TensorFlow只在CPU上运行的方法的主要内容，如果未能解决你的问题，请参考以下文章