tensorflow如何设置只在cpu上运行

Question

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

Answer 1

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

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

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

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

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

Answer 2

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

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

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

追答

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

本回答被网友采纳

Answer 3

这很简单，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