Не удалось создать дескриптор cudnn: CUDNN STATUS внутренняя ошибка - PullRequest
Новая
       4

Не удалось создать дескриптор cudnn: CUDNN STATUS внутренняя ошибка

1 голос
/ 06 мая 2019

Я пытаюсь создать машинное обучение в Python 3. но затем я пытаюсь скомпилировать свой код, я получил эту ошибку в Cuda 10.0 / cuDNN 7.5.0, кто-нибудь может мне помочь с этим?

RTX 2080

Я нахожусь на: Керас (2.2.4) tf-nightly-gpu (1.14.1.dev20190510)

Не удалось создать дескриптор cudnn: CUDNN_STATUS_INTERNAL_ERROR

Код ошибки: tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.

Вот мой код: 

model = Sequential()
model.add(Conv2D(32, (3, 3), activation='relu', input_shape=(50, 50, 1)))
model.add(MaxPooling2D((2, 2)))
model.add(Conv2D(64, (3, 3), activation='relu'))
model.add(MaxPooling2D((2, 2)))
model.add(Conv2D(64, (3, 3), activation='relu'))
model.add(Flatten())
model.add(Dense(1, activation='softmax'))

model.summary()

model.compile(optimizer='adam',
              loss='binary_crossentropy',
              metrics=['accuracy'])
model.fit(x, y, epochs=1, batch_size=n_batch)

OOM при выделении тензора с формой [24946,32,48,48] и типом float вкл / задание: локальный хост / реплика: 0 / задание: 0 / устройство: GPU: 0 по распределителю GPU_0_bfc

1 Ответ

0 голосов
/ 10 мая 2019

Есть 2 возможных решения.

Проблема с выделением памяти GPU

добавьте следующий код 

import tensorflow as tf
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.5)
config = tf.ConfigProto(gpu_options=gpu_options)
config.gpu_options.allow_growth = True
session = tf.Session(config=config)

проверьте также эту проблему

Проблема с драйвером NVIDIA

Как указано там вам необходимо обновить драйвер NVIDIA с помощью драйвера ODE.

Пожалуйста, проверьте Документация NVIDIA для версии драйвера

...