Как заставить cudnn работать?( ошибка инициализации ) - PullRequest
0 голосов
/ 20 декабря 2018

Получена эта ошибка в Windows 10

UnknownError: Не удалось получить алгоритм свертки.Вероятно, это связано с тем, что cuDNN не удалось инициализировать, поэтому попробуйте посмотреть, не было ли напечатано сообщение с предупреждением выше.[[{{node conv2d_1 / convolution}} = Conv2D [T = DT_FLOAT, _class = ["loc: @ training_1 / Adam / градиенты / conv2d_1 / convolution_grad / Conv2DBackpropFilter"], data_format = "NCHW", дилатации = [1, 1, 1, 1], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true, _device = "/ job: localhost / replica: 0 / task: 0 / device: GPU: 0"] (training_1 / Адам / градиенты / conv2d_1 / convolution_grad / Conv2DBackpropFilter-0-TransposeNHWCToNCHW-LayoutOptimizer, conv2d_1 / kernel / read)]] [[{{{loss loss_1 / mul / _267}} = _Recvclient_termined = false = «false», false: localhost / replica: 0 / task: 0 / device: CPU: 0 ", send_device =" / job: localhost / replica: 0 / task: 0 / device: GPU: 0 ", send_device_incarnation = 1, tenor_name =" edge_782_loss_1 /mul ", tenor_type = DT_FLOAT, _device =" / job: localhost / replica: 0 / task: 0 / device: CPU: 0 "]]

У меня RTx 2070 и:

Python 3.6.5 tf 1.12.0 tf-gpu 1.12.0 cuda 9.0 со всеми патчами.cudnn 7.3.1 keras 2.2.4

Я знаю страницу nvdia для cudnn и читаю некоторые другие ответы здесь.Мне интересны мелкие детали, которые отсутствуют.После перемещения 3 файлов в 3 директории в папке CUDA, есть еще один шаг?Возможно, есть порядок, в котором необходимо установить разные детали?

Кажется, что Cuda работает нормально, Python видит это, также Matlab видит это.

Произошла ошибка при запуске этого кода для mnist, полученного из Интернета, который работает, если я удаляю tenorflow-gpu и использую tenorflow на процессоре.

Примером большой помощи в прошлом было то, что вы не можете установить cuda, если вы не выберете пользовательский режим и не снимите флажок с опции visual studio.Спасибо !

Ответы [ 2 ]

0 голосов
/ 27 января 2019

Была похожая проблема с картой RTX 2070 с использованием CUDA 10 ...

Решение было использовать:

config.gpu_options.allow_growth = True

в тензорном потоке.

Подробнее о том, как использовать этот параметр: Как не допустить, чтобы тензорный поток выделял всю память GPU?

0 голосов
/ 07 января 2019

RTX-карт требуется CUDA 10 Я думаю.

...