GPU tensorflow运行问题

时间:2016-11-28 00:08:50

标签: tensorflow gpu

我已将GPU张量流安装为tutorial。相对环境是:

Ubuntu 14 + cuda8.0 + cudnn5.0 + GeForce Gt 640 (OEM)

问题是:

E tensorflow/stream_executor/cuda/cuda_driver.cc:491] failed call to cuInit: CUDA_ERROR_NO_DEVICE
I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:153] retrieving CUDA diagnostic information for host: wangyajie
I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:160] hostname: wangyajie
I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:185] libcuda reported version is: 352.63.0
I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:356] driver version file contents: """NVRM version: NVIDIA UNIX x86_64 Kernel Module  340.98  Mon Sep 19 17:31:03 PDT 2016
GCC version:  gcc version 4.8.4 (Ubuntu 4.8.4-2ubuntu1~14.04.3) 
"""
I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:189] kernel reported version is: 340.98.0
E tensorflow/stream_executor/cuda/cuda_diagnostics.cc:296] kernel version 340.98.0 does not match DSO version 352.63.0 -- cannot find working devices in this configuration

我该怎么做才能修复它? 我应该更新DSO吗?如果应该,应该使用什么命令?

如何查看天气我正确安装了cuda和cudnn?应该使用什么命令?

提前致谢! ZHQ

1 个答案:

答案 0 :(得分:0)

不知道这是怎么发生的,但是您的错误消息表明您的NVidia驱动程序内核模块与CUDA要求的版本不同。 (通常,如果您通过apt-get安装CUDA,它会检查此情况。)要解决此问题,您应该将驱动程序更新为352.63.0。说明可以在这里找到:

https://help.ubuntu.com/community/BinaryDriverHowto/Nvidia

但是,这可能无法完全解决您的问题,因为如果您单独安装它们,可能很难将CUDA和驱动程序放在同一版本上。相反,您应该考虑从同一个仓库安装两个,从而重新安装CUDA。

一般情况下,如果可能,我建议首先更新到Ubuntu 16.04,因为CUDA在那里安装的麻烦要少得多(根据我的经验,至少)。