2016-11-28 125 views
0

我已經安裝了GPU tensorflow作爲tutorial。相對環境是:GPU tensorflow運行問題

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

回答

0

不知道這是怎麼發生的,但是您的錯誤信息指出您的NVidia驅動程序內核模塊與CUDA需要的版本不同。 (通常,如果您通過apt-get安裝CUDA,它會檢查這一點。)要解決此問題,您應該將驅動程序更新爲352.63.0。說明可以在這裏找到:

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

然而,這可能無法完全解決您的問題,因爲它可能很難同時獲得CUDA和驅動程序安裝到相同的版本,如果您單獨安裝它們。相反,你應該考慮從同一個倉庫安裝兩個倉庫,從而重新安裝CUDA。

一般來說,如果可能的話,我建議你升級到Ubuntu 16.04首先,由於CUDA是痛苦要少得多,安裝有(在我的經驗,至少)。