Проблемы с установкой Theano с использованием CUDA при использовании пользователя без полномочий root

Question

Я следовал инструкциям по установке Theano GPUArray из исходных текстов (git-версий) в системные папки (не как пользователь). Тесты GPUArray работают без ошибок.

Проблема в том, что Theano работает только с GPU, если я запускаю с правами root. Выполнение примера для тестирования gpu:

(python35) rll@ip-30-92:~$ THEANO_FLAGS=device=cuda python temp.py 
ERROR (theano.gpuarray): Could not initialize pygpu, support disabled
Traceback (most recent call last):
  File "/usr/local/lib/python3.5/dist-packages/theano/gpuarray/__init__.py", line 179, in <module>
    use(config.device)
  File "/usr/local/lib/python3.5/dist-packages/theano/gpuarray/__init__.py", line 166, in use
    init_dev(device, preallocate=preallocate)
  File "/usr/local/lib/python3.5/dist-packages/theano/gpuarray/__init__.py", line 73, in init_dev
    context.cudnn_handle = dnn._make_handle(context)
  File "/usr/local/lib/python3.5/dist-packages/theano/gpuarray/dnn.py", line 83, in _make_handle
    cudnn = _dnn_lib()
  File "/usr/local/lib/python3.5/dist-packages/theano/gpuarray/dnn.py", line 70, in _dnn_lib
    raise RuntimeError('Could not find cudnn library (looked for v5* or v6*)')
RuntimeError: Could not find cudnn library (looked for v5* or v6*)
[Elemwise{exp,no_inplace}(<TensorType(float64, vector)>)]
Looping 1000 times took 3.201078 seconds
Result is [ 1.23178032  1.61879341  1.52278065 ...,  2.20771815  2.29967753
  1.62323285]
Used the cpu

При запуске от имени root он работает, хотя все еще существует ошибка, связанная с невозможностью идентификации устройств с помощью cuDNN:

(python35) rll@ip-30-92:~$ sudo THEANO_FLAGS=device=cuda python3 temp.py 
Can not use cuDNN on context None: cannot compile with cuDNN. We got this error:
b'/tmp/try_flags_bg7m03hd.c:4:19: fatal error: cudnn.h: No such file or directory\ncompilation terminated.\n'
Mapped name None to device cuda: TITAN X (Pascal) (0000:01:00.0)
[GpuElemwise{exp,no_inplace}(<GpuArrayType<None>(float64, vector)>), HostFromGpu(gpuarray)(GpuElemwise{exp,no_inplace}.0)]
Looping 1000 times took 0.390976 seconds
Result is [ 1.23178032  1.61879341  1.52278065 ...,  2.20771815  2.29967753
  1.62323285]
Used the gpu

Есть 2 Titan X на этой машине. Работает нормально с Tensorflow. Я не использую файл.theanorc, но я установил оба:

(python35) rll@ip-30-92:~$ echo $LD_LIBRARY_PATH 
/usr/local/cuda-8.0/lib64
(python35) rll@ip-30-92:~$ echo $CUDA_ROOT
/usr/local/cuda-8.0/

Я сделал все по инструкции, и, несмотря на некоторые предупреждения, ошибок не было.

Я не думаю, что это ошибка прав доступа в директории компиляции .theanoпотому что, если я чоун .theano Дир, поведение такое же.

Как я могу это исправить?

Answer 1

Я наконец нашел проблему. В инструкциях по установке Theano отсутствует аспект, который заключается в том, LIBRARY_PATH установить и добавить библиотеки CUDA к нему (обратите внимание, что это не LD_LIBRARY_PATH).

Если он не установлен, просто экспортируйте его, и вам будет хорошо идти. Итак, для временного исправления:

 export LIBRARY_PATH=/usr/local/cuda-8.0/lib64

Сохранение может зависеть от системы, но в целом вы можете добавить к /etc/environment, добавив строку:

LIBRARY_PATH=/usr/local/cuda-8.0/lib64

Это исправило сообщение, когда пользователь root, и исправило cuda для обычного пользователя.