小编Bra*_*Out的帖子

我在使用源编译的 Tensorflow 构建的 Keras 上运行卷积网络时遇到问题。我正在使用 CUDA 10.0 和 CuDNN 7.4，并且都已正确编译，如它们的示例生成文件所验证。当我运行 conv net 但不是密集网络时，我最终会收到这些错误：

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.
 [[{{node conv2d_1/convolution}}]]
 [[metrics/acc/Mean/_169]]

我尝试将 CUDA 和 CuDNN 的路径直接附加到我的路径中，尝试重新安装和重新编译 TensorFlow，但没有结果。版本冲突应该没有任何问题，因为这是在这台机器上全新安装。

import keras
import numpy as np
from keras.models import Sequential
from keras.layers import Dense, Flatten, Conv2D, MaxPooling2D, Dropout
from keras.datasets import mnist
import sys

def one_hot(data, num_categories):
    oh = np.zeros((len(data),num_categories))
    for …