Tensorflow GPU в Windows: CUDNN_STATUS_NOT_INITIALIZED - PullRequest
0 голосов
/ 02 сентября 2018

Я недавно обновил свою установку Tensorflow до Tensorflow-GPU и протестировал ее, запустив код из учебника, в котором была создана простая модель многоуровневой нейронной сети Perceptron. Все работало просто отлично, а выходной был идеальным. Затем я перешел к следующему уроку из этой серии (т. Е. К серии TensorFlow от Sentdex на YouTube), который изменил несколько строк кода для создания сверточной нейронной сети. Однако, когда я попытался запустить этот код, я получил следующую ошибку:

2018-09-02 03:17:30.563083: E T:srcgithubtensorflowtensorflowstream_executorcudacuda_dnn.cc:352] Could not create cudnn handle: CUDNN_STATUS_NOT_INITIALIZED
2018-09-02 03:17:30.563336: E T:srcgithubtensorflowtensorflowstream_executorcudacuda_dnn.cc:356] Error retrieving driver version: Unimplemented: kernel reported driver version not implemented on Windows

Быстрый переход к коду MLP подтвердил, что он все еще отлично работает на моем GPU, но по какой-то причине код ConvNN выбрасывает вышеупомянутые ошибки.

У кого-нибудь есть идея, почему Tensorflow-GPU работает для модели MLP, но не для модели ConvNet? Что это за драйвер, который, кажется, отсутствует или сломан? Что случилось с CUDNN не инициализирован? Я приветствую любые комментарии или комментарии по теме. Если эта проблема уже была рассмотрена в другом месте, я приношу свои извинения и был бы очень признателен за то, что вы перешли на соответствующую ветку. Я искал несколько разных способов ответа и не увидел ничего, связанного с моей конкретной проблемой.

Заранее спасибо за ваше время и помощь. Патрик.

P.S. Чтобы предоставить полную информацию, я использую TensorFlow-gpu v.1.10.0, набор инструментов CUDA v 9.0, CuDNN v 7.2.1 через Pycharm на ноутбуке Lenovo Yoga14 (corei5 @ 2.3 ГГц, 8 ГБ ОЗУ, 256 ГБ SSD) , Он имеет графический процессор Nvidia 940M с 2 ГБ видеопамяти. Мой интерпретатор Python - Anaconda 3.6.5.

...