tensorflow使用cpu不用gpu的问题

前提：cuda版本和cudnn版本已安装好，版本对应没问题；

问题：在安装好cuda、tensorflow、tensorflow-gpu的环境下，调用tensorflow跑模型，发现cpu资源占满，而gpu处于空闲状态。

导致上述问题的原因可能有两个：

（1）tensorflow-gpu的版本低于tensorflow的版本；
（2）cuda版本与tensorflow-gpu版本对应不上；

对于第一个原因：tensorflow-gpu的版本低于tensorflow的版本，有以下检查方法：

（1）跑模型，然后bash环境输入nvidia-smi直接查看是否正使用gpu（也能查看CUDA版本）

（2）在python环境下打印设备信息

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

1）假如tensorflow-gpu版本低于tensorflow版本时，只会输出以下设备全是cpu的信息：

2）假如tensorflow-gpu版本高于tensorflow版本时，会同时输出cpu和gpu信息：

所以，如果是tensorflow-gpu版本低于tensorflow版本的问题，那直接更新下tensorflow-gpu的版本，使其大于tensorflow版本。

对于第二个原因，cuda版本与tensorflow-gpu版本对应不上，有以下检查方法：

版本对应关系

（1）查看cudnn版本：cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

下面显示cudnn版本信息是7.6.5

（2）查看cuda版本：nvcc -V 或 nvcc --version 或 nvidia-smi

所以，如果是tensorflow-gpu版本和cuda版本对应的问题，同样也是更新tensorflow-gpu的版本，并且保证更新后的版本大于tensorflow版本。

tensorflow使用cpu不用gpu的问题相关推荐

TensorFlow指定CPU和GPU方法
TensorFlow指定CPU和GPU方法 TensorFlow 支持 CPU 和 GPU.它也支持分布式计算.可以在一个或多个计算机系统的多个设备上使用 TensorFlow. TensorFlow ...
tensorflow 制定 CPU 或GPU
tensorflow 制定 CPU 或GPU CPU with tf.Session() as sess:...: with tf.device("/cpu:0"):...: m1 ...
TensorFlow测试CPU、GPU
目录 1.查看当前Tensorflow版本 2.查看当前主机运行的设备 3.查看GPU是否可用,指定在CPU/GPU运行 4.比较CPU和GPU上的运行时间 1.查看当前Tensorflow版本 im ...
Tensorflow 使用cpu和gpu的区别
在Tensorflow中使用gpu和cpu是有很大的差别的.在小数据集的情况下,cpu和gpu的性能差别不大.不过在大数据集的情况下,cpu的时间显著增加,而gpu变化并不明显.不过,我的笔记本电脑的 ...
tensorflow的CPU和GPU的区别
最近深度学习CPU根本不够用,老板配置了GPU 怎么能不会用呢? 原来从来没有关注过的底层东西,于是乎痛下决心搞清楚原理和操作先来看看第一部分为什么GPU比CPU更diao呢? 这里就需要从他么的 ...
解决linux的-bash: ./xx: Permission denied/tensorflow 运行cpu还是gpu的方法
在linux下执行sh文件时提示下面信息: -bash: ./xx.sh: Permission denied 解决: chmod 777 xx.sh 今天在公司的服务器上用tensorflow训练, ...
Tensorflow选择CPU或GPU运行
代码 import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUD ...
win10和Ubuntu16.04中基于anaconda安装TensorFlow（CPU+GPU）
目录 Windows系统下安装步骤安装CPU版本的tensorflow 安装GPU版本的tensorflow(使用GPU加速的tensorflow模型训练速度更快) Ubuntu中的安装步骤安装C ...
python调用gpu进行运算_tensorflow指定CPU与GPU运算的方法实现
1.指定GPU运算如果安装的是GPU版本,在运行的过程中TensorFlow能够自动检测.如果检测到GPU,TensorFlow会尽可能的利用找到的第一个GPU来执行操作. 如果机器上有超过一个可用 ...

tensorflow使用cpu不用gpu的问题

tensorflow使用cpu不用gpu的问题相关推荐

最新文章

热门文章