tensorflow - list_local_device tensorflow 未检测到 gpu

Question

有没有办法检查我安装了 GPU 的 Tensorflow 版本？

!nvidia-smi

2017 年 12 月 18 日星期一 23:58:01

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 384.90                 Driver Version: 384.90                    |
|-------------------------------+----------------------+----------------------+
| 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  | 00000000:01:00.0  On |                  N/A |
| N/A   53C    P0    31W /  N/A |   1093MiB /  8105MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      1068      G   /usr/lib/xorg/Xorg                           599MiB |
|    0      2925      G   compiz                                       290MiB |
|    0      3611      G   ...-token=11A9F5872A56620B72D1D5DF707CF1FC   200MiB |
|    0      5786      G   /usr/bin/nvidia-settings                       0MiB |
+-----------------------------------------------------------------------------+

但是当我尝试检测列表本地设备时，只检测到 CPU。

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

[name: "/cpu:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 3303842605833347443
]

我是否必须设置其他内容才能将 GPU 用于 Keras 或 Tensorflow ？

Answer 1

使用 pip install tensorflow-gpu或 conda install tensorflow-gpu对于 tensorflow 的 GPU 版本。如果您使用的是 keras-gpu conda install -c anaconda keras-gpu命令将自动安装 tensorflow-gpu 版本。在执行这些任何命令之前，请确保您卸载了正常的 tensorflow 。