安装CUDA工具包和cuDNN后,Tensorflow无法识别GPU

时间:2017-05-25 01:07:46

标签: python tensorflow

我有一个1070 gtx。我跑的时候:

> sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

我得到了

> 2017-05-24 21:01:07.959988: W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.1 instructions, but these are available on your machine and could speed up CPU computations.
> 2017-05-24 21:01:07.960018: W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.2 instructions, but these are available on your machine and could speed up CPU computations.
> 2017-05-24 21:01:07.960030: W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use AVX instructions, but these are available on your machine and could speed up CPU computations.
> Device mapping: no known devices.
> 2017-05-24 21:01:07.962291: I tensorflow/core/common_runtime/direct_session.cc:257] Device mapping:

来自nvidia-smi的结果:

Wed May 24 21:47:25 2017
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 375.39                 Driver Version: 375.39                    |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 1070    Off  | 0000:02:00.0      On |                  N/A |
|  0%   44C    P8     8W / 180W |    354MiB /  8112MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID  Type  Process name                               Usage      |
|=============================================================================|
|    0      1179    G   /usr/lib/xorg/Xorg                             184MiB |
|    0      2343    G   compiz                                         120MiB |
|    0     26145    G   ...el-token=A9030FB8E905376BBEE74D68F09DFE81    47MiB |
+-----------------------------------------------------------------------------+

这告诉我tensorflow没有使用GPU。 我安装了CUDA 8.0,cuDNN 5.1和tensorflow 1.1.0。我不确定是什么问题。请帮忙。

2 个答案:

答案 0 :(得分:2)

正如@Alexander Yau指出的那样,卸载常规的tensoflow软件包就可以了。谢谢。

答案 1 :(得分:0)

你安装了nvidia图形驱动程序吗?它也是必需的。

您可以通过以下方式测试是否安装了驱动程序:

$ nvidia-smi