使用 keras 训练 CNN model 可与 CPU 一起使用，但不适用于 GPU

Question

我正在训练 CNN model 对简单图像（正方形和十字）进行分类，当我使用 cpu 时一切正常，但是当我使用 gpu 时，一切正常，直到训练开始，我得到这个错误：

2022-06-15 04:25:49.158944：我 tensorflow/stream_executor/cuda/cuda_dnn.cc:384] 加载 cuDNN 版本 8401

然后程序就停止了。 有谁知道如何解决这个问题？

Answer 1

如果您使用 pycharm，您可以 select 的“在 output 控制台中模拟终端”选项打印详细的错误信息。

运行->编辑配置->执行->在 output 控制台中模拟终端

也许 CUDA 缺少 zlibwapi.dll 文件，您可以下载它并将其移动到 cuda 的 bin 中。