Theano - pyGpuArray не может найти libnvrtc.so

Я недавно изменил конфигурацию, чтобы тренировать CNN быстрее. Я получил GTX 1080. Для большинства моих существующих нейронных сетей я использую theano back-end через python3.

Поскольку я прочитал, что архитектура pascal не была полностью использована в CUDA 7.5, я установил CUDA 8 и соответствующий cudnn 6.X.

Моя проблема заключается в следующем, Pygpuarray, похоже, не может найти "libnvrtc.so":

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 65, in init_dev
    sched=config.gpuarray.sched)
  File "pygpu/gpuarray.pyx", line 634, in pygpu.gpuarray.init (pygpu/gpuarray.c:9428)
  File "pygpu/gpuarray.pyx", line 584, in pygpu.gpuarray.pygpu_init (pygpu/gpuarray.c:9119)
  File "pygpu/gpuarray.pyx", line 1057, in pygpu.gpuarray.GpuContext.__cinit__ (pygpu/gpuarray.c:13421)
pygpu.gpuarray.GpuArrayException: b'Could not load "libnvrtc.so": libnvrtc.so: cannot open shared object file: No such file or directory'

Я почти уверен, что cuda установлена, вот вывод nvcc --version:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2016 NVIDIA Corporation
Built on Tue_Jan_10_13:22:03_CST_2017
Cuda compilation tools, release 8.0, V8.0.61

И я нашел файл в: /usr/local/cuda-8.0/lib64

Странно то, что команда "ldconfig -p | grep libnvcc" ничего не возвращает...

Я надеюсь, что этот пост не существует где-либо еще, я сделал несколько поисков, чтобы предотвратить это, но вы знаете, Интернет огромен.

Спросите меня, если вы хотите больше информации. Поскольку мне не удалось решить эту проблему, я, очевидно, не знаю, какая информация необходима для решения этой проблемы, поэтому я сделал это небольшое резюме, ожидая ваших вопросов (при необходимости).

0 ответов

Другие вопросы по тегам