python - Tensorflow Keras 错误“找不到算法！”

Question

我目前正在关注使用 Scikit-Learn、Keras 和 TensorFlow 进行机器学习实践这本书，但我不断遇到这些错误。当我运行以下代码时：

model.compile(loss="sparse_categorical_crossentropy", optimizer="nadam", metrics=["accuracy"])
history = model.fit(X_train, y_train, epochs=10, validation_data=(X_valid, y_valid))
score = model.evaluate(X_test, y_test)
y_pred = model.predict(X_new)

我收到这个错误

NotFoundError：没有算法有效！[[节点顺序/conv2d/Conv2D（定义于：2）]] [Op:__inference_train_function_2275]

函数调用栈：train_function

然后我从堆栈溢出中提取了这两件事，这似乎“解决”了问题

os.environ['TF_CPP_MIN_LOG_LEVEL'] = "2"

config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)

但是，此消息会阻塞我的终端：

2021-04-10 18:20:01.643838: W tensorflow/stream_executor/gpu/asm_compiler.cc:235] 您的 CUDA 软件堆栈已旧。我们回退到 NVIDIA 驱动程序进行一些编译。更新您的 CUDA 版本以获得最佳性能。ptxas 错误为：ptxas 致命：未为选项“gpu-name”定义值“sm_86”

我正在运行 cuda 11.2，并且我的驱动程序版本为 460.39。我的卡是 3080。有谁知道可能出了什么问题？

标签： pythonkerastensorflow2.0