Tensorflow 未在 GPU 上运行 [英] Tensorflow not running on GPU
问题描述
我已经花费了大量时间在堆栈溢出问题上进行挖掘并寻找答案，但一无所获
大家好，
我在上面运行带有 Keras 的 Tensorflow.我 90% 确定我安装了 Tensorflow GPU，有什么方法可以检查我安装的是哪个?
I am running Tensorflow with Keras on top. I am 90% sure I installed Tensorflow GPU, is there any way to check which install I did?
我试图从 Jupyter notebook 运行一些 CNN 模型，我注意到 Keras 在 CPU 上运行模型(检查任务管理器，CPU 处于 100%).
I was trying to do run some CNN models from Jupyter notebook and I noticed that Keras was running the model on the CPU (checked task manager, CPU was at 100%).
我尝试从 tensorflow 网站运行此代码:
I tried running this code from the tensorflow website:
# Creates a graph.
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
# Creates a session with log_device_placement set to True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
# Runs the op.
print(sess.run(c))
这就是我得到的:
MatMul: (MatMul): /job:localhost/replica:0/task:0/cpu:0
2017-06-29 17:09:38.783183: I c: f_jenkinshomeworkspace
elease-winmwindowspy35 ensorflowcorecommon_runtimesimple_placer.cc:847] MatMul: (MatMul)/job:localhost/replica:0/task:0/cpu:0
b: (Const): /job:localhost/replica:0/task:0/cpu:0
2017-06-29 17:09:38.784779: I c: f_jenkinshomeworkspace
elease-winmwindowspy35 ensorflowcorecommon_runtimesimple_placer.cc:847] b: (Const)/job:localhost/replica:0/task:0/cpu:0
a: (Const): /job:localhost/replica:0/task:0/cpu:0
2017-06-29 17:09:38.786128: I c: f_jenkinshomeworkspace
elease-winmwindowspy35 ensorflowcorecommon_runtimesimple_placer.cc:847] a: (Const)/job:localhost/replica:0/task:0/cpu:0
[[ 22. 28.]
[ 49. 64.]]
这对我来说表明我出于某种原因在我的 CPU 上运行.
Which to me shows I am running on my CPU, for some reason.
我有一个 GTX1050(驱动程序版本 382.53)，我安装了 CUDA 和 Cudnn，并且安装了 tensorflow，没有任何问题.我也安装了 Visual Studio 2015，因为它被列为兼容版本.
I have a GTX1050 (driver version 382.53), I installed CUDA, and Cudnn, and tensorflow installed without any problems. I installed Visual Studio 2015 as well since it was listed as a compatible version.
我记得 CUDA 提到过安装不兼容的驱动程序，但如果我没记错的话，CUDA 应该已经安装了自己的驱动程序.
I remember CUDA mentioning something about an incompatible driver being installed, but if I recall correctly CUDA should have installed its own driver.
我运行了这些命令来列出可用的设备
I ran theses commands to list the available devices
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
这就是我得到的
[name: "/cpu:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 14922788031522107450
]
还有很多这样的警告
2017-06-29 17:32:45.401429: W c: f_jenkinshomeworkspace
elease-winmwindowspy35 ensorflowcoreplatformcpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE instructions, but these are available on your machine and could speed up CPU computations.
编辑 2
尝试运行
pip3 install --upgrade tensorflow-gpu
我明白了
Requirement already up-to-date: tensorflow-gpu in c:usersxxxappdatalocalprogramspythonpython35libsite-packages
Requirement already up-to-date: markdown==2.2.0 in c:usersxxxappdatalocalprogramspythonpython35libsite-packages (from tensorflow-gpu)
Requirement already up-to-date: html5lib==0.9999999 in c:usersxxxappdatalocalprogramspythonpython35libsite-packages (from tensorflow-gpu)
Requirement already up-to-date: werkzeug>=0.11.10 in c:usersxxxappdatalocalprogramspythonpython35libsite-packages (from tensorflow-gpu)
Requirement already up-to-date: wheel>=0.26 in c:usersxxxappdatalocalprogramspythonpython35libsite-packages (from tensorflow-gpu)
Requirement already up-to-date: bleach==1.5.0 in c:usersxxxappdatalocalprogramspythonpython35libsite-packages (from tensorflow-gpu)
Requirement already up-to-date: six>=1.10.0 in c:usersxxxappdatalocalprogramspythonpython35libsite-packages (from tensorflow-gpu)
Requirement already up-to-date: protobuf>=3.2.0 in c:usersxxxappdatalocalprogramspythonpython35libsite-packages (from tensorflow-gpu)
Requirement already up-to-date: backports.weakref==1.0rc1 in c:usersxxxappdatalocalprogramspythonpython35libsite-packages (from tensorflow-gpu)
Requirement already up-to-date: numpy>=1.11.0 in c:usersxxxappdatalocalprogramspythonpython35libsite-packages (from tensorflow-gpu)
Requirement already up-to-date: setuptools in c:usersxxxappdatalocalprogramspythonpython35libsite-packages (from protobuf>=3.2.0->tensorflow-gpu)
已解决:检查解决方案的评论.感谢所有帮助过的人！
Solved: Check comments for solution. Thanks to all who helped!
我是新手，所以非常感谢任何帮助！谢谢.
I am new to this, so any help is greatly appreciated! Thank you.
推荐答案
要检查哪些设备可用于 TensorFlow，您可以使用它并查看 GPU 卡是否可用:
To check which devices are available to TensorFlow you can use this and see if the GPU cards are available:
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
编辑此外，如果您使用 TensorFlow Cuda 版本，您应该会看到此类日志:
Edit Also, you should see this kind of logs if you use TensorFlow Cuda version :
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcublas.so.*.* locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcudnn.so.*.* locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcufft.so.*.* locally
这篇关于Tensorflow 未在 GPU 上运行的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！
