为什么 keras model.fit_generator() 显示错误，例如无法获取卷积算法?

Question

tensorflow version-> 2.0.0

keras version--> 2.2.4

datagen = ImageDataGenerator(
featurewise_center=False,  # set input mean to 0 over the dataset
samplewise_center=False,  # set each sample mean to 0
featurewise_std_normalization=False,  # divide inputs by std of the dataset
samplewise_std_normalization=False,  # divide each input by its std
zca_whitening=False,  # apply ZCA whitening
rotation_range=10,  # randomly rotate images in the range (degrees, 0 to 180)
zoom_range = 0.1, # Randomly zoom image 
width_shift_range=0.1,  # randomly shift images horizontally (fraction of total width)
height_shift_range=0.1,  # randomly shift images vertically (fraction of total height)
horizontal_flip=False,  # randomly flip images
vertical_flip=False)  # randomly flip images
datagen.fit(X_train)

history = model.fit_generator(datagen.flow(X_train,Y_train,batch_size=batch_size,),
epochs=epochs,validation_data=(X_val,Y_val),
verbose=2,steps_per_epoch=X_train.shape[0] // batch_size,
callbacks=[learning_rate_reduction])

错误：

未知错误：无法获取卷积算法。这可能是 因为 cuDNN 初始化失败，所以请尝试查看是否有警告 日志消息打印在上面。[操作：Conv2D]

Answer 1

试试这对我有用

库达版本 --> 10.0 库德恩版本 --> 7.6.4

张量流 GPU 版本 --> 2.0.0 Keras GPU 版本 --> '2.2.4'