2016-11-05 168 views
4

我是Ubuntu的新用戶,我花了一個多星期的時間嘗試安裝gpu tensorflow。我試圖遵循網站和博客的教程。但是每當我在我的python控制檯中導入tensorflow。它總是顯示:GPU tensorflow安裝問題

import tensorflow 
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcublas.so.8.0 locally 
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcudnn.so.5.1.5 locally 
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcufft.so.8.0 locally 
I tensorflow/stream_executor/dso_loader.cc:119] Couldn't open CUDA library libcuda.so.1. LD_LIBRARY_PATH: :/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64 
I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:165] hostname: HanGuo 
I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:189] libcuda reported version is: Not found: was unable to find libcuda.so DSO loaded into this program 
I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:193] kernel reported version is: Permission denied: could not open driver version path for reading: /proc/driver/nvidia/version 
I tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:1091] LD_LIBRARY_PATH: :/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64 
I tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:1092] failed to find libcuda.so on this system: Failed precondition: could not dlopen DSO: libcuda.so.1; dlerror: libnvidia-fatbinaryloader.so.367.57: cannot open shared object file: No such file or directory 
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcurand.so.8.0 locally 

我試圖尋找答案全谷歌,但沒有真正的解決辦法想出了yet.Can有人幫我嗎?

+0

給我們提供更多關於您如何安裝Tensorflow和Cuda的信息。 – BernardoGO

+0

我從Nvidia網站下載CUDA.run文件,並使用'$ sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz'將012安裝到'/ usr/local/cuda'中。 '$ sudo cp cuda/include/cudnn.h/usr/local/cuda/include'$ sudo cmod cda/lib64/libcudnn */usr/local/cuda/lib64' '$ sudo chmod a + r/usr/local/cuda/include/cudnn.h''/ usr/local/cuda/lib64/libcudnn *'。 tensorflow使用'git clone https:// github.com/tensorflow/tensorflow'下載,'。/ configure' –

+0

基本上我遵循(https://alliseesolutions.wordpress.com/2016/09/08/install-gpu-tensorflow-from-sources -w-ubuntu-16-04-and-cuda-8-0-rc /)說。 –

回答

2

根據$ ls /usr/local/cuda的輸出,您已成功安裝它。

現在,什麼是缺少PATH定義,以便TensorFlow可以找到它在哪裏。

1)打開你的.bashrc文件:gedit ~/.bashrc

2)追加以下兩行到文件的末尾:

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/include" 
export CUDA_HOME=/usr/local/cuda 

3)一旦保存並關閉文本文件就可以了返回到原來的終端輸入這個命令來重新加載你的.bashrc文件:

source ~/.bashrc 

雖然第三部分是應該採取照顧,如果你想成爲100%確保它能夠重新啓動計算機。

現在,對於TensorFlow,請記住將CUDA路徑設置爲/usr/local/cuda,並將版本設置爲8.0。

-1

安裝libcuda5-dev可能會解決您的問題