Допустимый код в ядре Kaggle: UnknownError: Не удалось получить алгоритм свертки. Вероятно, это связано с тем, что cuDNN не удалось инициализировать - PullRequest
Новая
       9

Допустимый код в ядре Kaggle: UnknownError: Не удалось получить алгоритм свертки. Вероятно, это связано с тем, что cuDNN не удалось инициализировать

0 голосов
/ 01 ноября 2019

Когда я запускаю следующий код в ядре Kaggle, я получаю 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., но если я запускаю его локально, он работает локально. 

!pip install --upgrade tensorflow_addons
!pip install --upgrade tensorflow_datasets

import tensorflow as tf
import numpy as np

from tensorflow.keras.layers import Conv2D
from tensorflow.keras.models import Sequential

model = Sequential()
model.add(Conv2D(20, kernel_size=(2,2), input_shape=(256, 256, 3)))
model.compile(loss='mse', optimizer=tf.keras.optimizers.Adam(lr=0.002))

model.predict(np.random.random((1, 256, 256, 3)))

Код кажется совершенно корректным, почему возникает ошибка?

РЕДАКТИРОВАТЬ: эта ошибка возникает только при включенном графическом процессоре.

...