Как использовать другой пакет cudnn, отличный от пакета по умолчанию в Anaconda - PullRequest
Новая
       15

Как использовать другой пакет cudnn, отличный от пакета по умолчанию в Anaconda

0 голосов
/ 20 апреля 2019

У меня есть установка anaconda, в которой я настроил python 3.6, tenensflow 1.13 на windows 10. Установка работает нормально, за исключением случаев, когда я пытаюсь вызвать код cudnn, то есть conv2d и аналогичные функции, когда я встретил следующую ошибку: «E tensorflow / stream_executor/cuda/cuda_dnn.cc:324] Загруженная библиотека CuDNN времени выполнения: 7.3.1, но источник был скомпилирован с: 7.4.1 Основная и дополнительная версия библиотеки CuDNN должны совпадать или иметь более высокую младшую версию в случае CuDNN 7.0 или более поздней версии.При использовании двоичной установки обновите библиотеку CuDNN ".

Какие комбинации версий TensorFlow и CUDA совместимы? .Из этого поста я попытался установить cudnn 7.4.1 на свой рабочий стол, но похоже, что anaconda продолжает использовать только cudnn 7.3.1 по умолчанию и не ссылается на тот, на который указывают мои переменные пути.Буду очень признателен за любую помощь в том, как я могу заставить conda использовать cudnn 7.4.1.

1 Ответ

0 голосов
/ 27 июня 2019

Я не знаю, была ли решена проблема оригинального плаката. На случай, если у кого-то возникнет такая же проблема (как и у меня), сначала нужно попробовать это:

https://towardsdatascience.com/tensorflow-gpu-installation-made-easy-use-conda-instead-of-pip-52e5249374bc

По существу, он рекомендует вам напрямую conda tensorflow-gpu и позволить conda обрабатывать версии cudnn и cuda. По крайней мере после этого, если вы conda list отобразите обновленную (или более актуальную) cudnn версию.

Но в моем случае после этого у меня все еще была та же ошибка. Поэтому я немного покопался и обнаружил, что это было из одного из моих других пакетов. В папке anaconda вы можете найти все вхождения cudnn64_7.dll. Для меня это пакет mxnet-cu100, который (даже с самой новой версией) все еще использует более старую версию cudnn dll. Я просто скопировал более новую cudnn64_7.dll в эту папку site-packages (думаю, с некоторым риском, но на этом этапе мне было нечего терять), и это решило проблему.

...