tensorflow如何设置只在cpu上运行

Posted 2023-02-25

with tf.device("/cpu:0")使用后，还是在gpu上运行，怎么设置不适用gpu，着急用，求大神，谢谢

1、首先在win10上点击开始菜单，直接输入cmd，能够查询出命令提示符。

2、右键命令提示符，以管理员身份运行。

3、打开命令提示符窗口，输入pip install --upgrade --ignore-installed tensorflow开始安装tensorflow。

4、安装过程中会下载很多相关的依赖安装包，安装完成如下图所示。

5、输入python ，然后输入下图所示的代码，有下图回显说明tensorflow安装成功了。

参考技术A with tf.Session() as sess:
with tf.device("/cpu:0"):
后面写你的代码

如果是新手的话，建议安装tensorflow而不是tensorflow-gpu
希望可以帮到你追问

你好，这个好像还是在gpu上

追答

你那个是最开始tensorflow框架的错误，应该安装tensorflow而不是tensorflow-gpu

参考技术B

这很简单，tensorflow-gpu是使用CUDA的

设置CUDA参数为-1就行，-1就是不使用GPU

代码的最後一行就是了，建议全部复制贴上

import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='3' #Disable Tensorflow debugging information
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ['CUDA_VISIBLE_DEVICES'] = '-1' #Set GPU device -1

限制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"