我有一个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。我不确定是什么问题。请帮忙。
答案 0 :(得分:2)
正如@Alexander Yau指出的那样,卸载常规的tensoflow软件包就可以了。谢谢。
答案 1 :(得分:0)
你安装了nvidia图形驱动程序吗?它也是必需的。
您可以通过以下方式测试是否安装了驱动程序:
$ nvidia-smi