2016-11-05 2 views
4

Я новый пользователь Ubuntu, и я потратил более недели на установку gpu tensorflow. Я попытался следить за учебниками с сайта и блогов. Но всякий раз, когда я импортирую shadoworflow в свою консоль python. Он всегда показывает:Ошибка установки 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 

Я пытался найти ответ во всем Google, но никакого реального решения не придумал yet.Can кто-нибудь помочь мне?

+0

Дайте нам больше информации о том, как вы установили Tensorflow и Cuda. – BernardoGO

+0

Загрузите файл CUDA .run с веб-сайта Nvidia и установите cuda в: '/ usr/local/cuda', используя' $ sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz' '$ sudo cp cuda/include/cudnn.h/usr/local/cuda/include' '$ sudo cp cuda/lib64/libcudnn */usr/local/cuda/lib64' ' $ sudo chmod a + r/usr/local/cuda/include/cudnn.h''/usr/local/cuda/lib64/libcudnn * '. Тензорный поток загружался с использованием '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 может решить вашу проблему

Смежные вопросы