2016-04-18 2 views
1

Я использую Python с IDE Pycharm в системе Ubuntu с установленным и работающим CUDA 7.5.Ошибка CudNN TensorFlow: Не удалось установить дескриптор фильтра cudnn: CUDNN_STATUS_BAD_PARAM

Я только что объединил файлы CUDNN с моей общей установкой CUDA.

Теперь, когда я бегу глубоко сверточного Nn учебник с сайта TensorFlow, я получаю следующее сообщение об ошибке:

F tensorflow/stream_executor/cuda/cuda_dnn.cc:383] could not set cudnn filter descriptor: CUDNN_STATUS_BAD_PARAM

Кто-нибудь знает, что источник этой проблемы? Это проблема с моей установкой CUDNN? (который был shisty - я только побежал rsync -av ./include/ /usr/local/cuda/include и rsync -av ./lib64/ /usr/local/cuda/lib64, тогда я re-ran ldconfig /usr/local/cuda/lib64) ... или опечатка в коде?


Или это может быть откуда-то еще? (или, даже лучше, кто-нибудь знает об исправлении?)

ответ

4

Хорошо, я столкнулся с нулевыми ответами в Интернете, поэтому я собираюсь оставить это на случай, если кто-то другой столкнется с той же проблемой.


Проблема в том, что cudnn v5 не работал с тензором потока (может быть, в общем, может быть, только на моем установить ..., который, по общему признанию, является Ubuntu 15,10 вместо 15,04 поддерживается CUDA), так что я должен был вернуться к cudnn v4.

Предполагая, что вы установили CUDA 7.5 в каталог по умолчанию и настроили символическую ссылку в каталоге/usr/local/cuda, просто используя CUDNN v4 (с февраля), он будет работать. Я просто извлек деготь файл CUDNN и побежал следующие команды, с корневыми разрешениями (из моих загрузки папок):

tar -xvzf cudnn-etc.tgz 
cd cuda 

rsync -av ./lib64/ /usr/local/cuda-7.5/lib64/ 
rsync -av ./include/ /usr/local/cuda-7.5/lib64/ 

ldconfig /usr/local/cuda/lib64 
+0

Я столкнулся с тем же выпуском 16.04 LTS. Обратите внимание, что если вы скомпилируете TF из исходного кода, теперь вы можете использовать cudnn v5. Однако версия протокола TF по-прежнему (по состоянию на июль '16) застряла в cudnn v4. Кажется, что люди с GPU Pascal должны установить из источника, поскольку Pascal поддерживает только cudnn v5 или новее. – mastazi

1

Вы используете версию 5 из cudnn

См Github issue #1787

Патч #1794 объединяется в master, поэтому последний RC-тензор (r0.8) has support for it.

+0

Хорошо, спасибо, спасибо. Я обновлю, как только закончу проект (теперь все работает, не прикасаюсь к нему :) – bordeo

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