CUDA，CuDNN安装，但Tensorflow不能使用GPU

我的系统是Ubuntu的14.04上EC2：CUDA，CuDNN安装，但Tensorflow不能使用GPU

nvidia-smi 
Sun Oct 2 13:35:28 2016  
+------------------------------------------------------+      
| NVIDIA-SMI 352.63  Driver Version: 352.63   |      
|-------------------------------+----------------------+----------------------+ 
| GPU Name  Persistence-M| Bus-Id  Disp.A | Volatile Uncorr. ECC | 
| Fan Temp Perf Pwr:Usage/Cap|   Memory-Usage | GPU-Util Compute M. | 
|===============================+======================+======================| 
| 0 GRID K520   Off | 0000:00:03.0  Off |     N/A | 
| N/A 37C P0 35W/125W |  11MiB/4095MiB |  0%  Default | 
+-------------------------------+----------------------+----------------------+ 

+-----------------------------------------------------------------------------+ 
| Processes:              GPU Memory | 
| GPU  PID Type Process name        Usage  | 
|=============================================================================| 
| No running processes found             | 
+-----------------------------------------------------------------------------+ 
[email protected]:~$ nvcc -V 
nvcc: NVIDIA (R) Cuda compiler driver 
Copyright (c) 2005-2015 NVIDIA Corporation 
Built on Tue_Aug_11_14:27:32_CDT_2015 
Cuda compilation tools, release 7.5, V7.5.17

我安装CUDA 7.5和5.1 CuDNN。

我在/ usr/local/local/lib64中包含正确的文件并包含文件夹。

Tensorflow行给出什么：

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

>>> sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) 
Device mapping: no known devices. 
I tensorflow/core/common_runtime/direct_session.cc:252] Device mapping: 

>>>

请帮忙（感谢很多:)）。

来源

2016-10-02 John77

@talonmies现有的解决方案是n没有工作，我一次又一次检查了路径变量。 – John77

您似乎有一个正在运行的CUDA安装。因此，这个问题是一个张量配置问题，与CUDA编程无关。这就是我回收标签的原因。 – talonmies

@talonmies是的，多数民众赞成我认为，必须有一些问题在链接或路径......就像那样。你能提出一些我可以尝试的建议吗？谢谢。 – John77

你是如何构建张量流的？

如果你是用bazel做的，你是否正确添加--config = cuda？

如果你使用pip安装它，你是否正确地使用了gpu enable？

编辑：

您可以在这里看到如何使用PIP安装： https://www.tensorflow.org/versions/r0.11/get_started/os_setup.html#pip-installation

你需要采取一个二进制兼容GPU：

# Ubuntu/Linux 64-bit, GPU enabled, Python 2.7 
# Requires CUDA toolkit 7.5 and CuDNN v5. For other versions, see "Install from sources" below. 
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc0-cp27-none-linux_x86_64.whl 

# Mac OS X, GPU enabled, Python 2.7: 
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/mac/gpu/tensorflow-0.11.0rc0-py2-none-any.whl 

# Ubuntu/Linux 64-bit, GPU enabled, Python 3.4 
# Requires CUDA toolkit 7.5 and CuDNN v5. For other versions, see "Install from sources" below. 
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc0-cp34-cp34m-linux_x86_64.whl 

# Ubuntu/Linux 64-bit, GPU enabled, Python 3.5 
# Requires CUDA toolkit 7.5 and CuDNN v5. For other versions, see "Install from sources" below. 
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc0-cp35-cp35m-linux_x86_64.whl 

# Mac OS X, GPU enabled, Python 3.4 or 3.5: 
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/mac/gpu/tensorflow-0.11.0rc0-py3-none-any.whl

然后安装tensorflow：

# Python 2 
$ sudo pip install --upgrade $TF_BINARY_URL 

# Python 3 
$ sudo pip3 install --upgrade $TF_BINARY_URL

来源

2016-10-03 03:27:59 rAyyy

gpu的pip选项是什么？我没有看到任何gpu命令alonside pip。请让我知道是否有任何命令，我会将其标记为答案。谢谢。 – John77

我编辑了我的答案，你可以检查它。我希望它有帮助。没有真正的选择只需要采取正确的二进制。 – rAyyy

CUDA，CuDNN安装，但Tensorflow不能使用GPU

回答

相关问题