Keras 需要很长时间才能拟合模型

Question

我正在从here 运行一个简单的 mnist python 脚本

程序似乎停在这一行：

model.fit(x_train, y_train, batch_size=batch_size, epochs=epochs, validation_split=0.1)

enter image description here 大约需要 >=30 分钟 ...

有人说是GPU内存here。

我正在使用 RTX 3080、cuda 11.2、cudnn 7.6，有人可以帮帮我吗？

Answer 1

RTX 3080 卡基于Ampere 架构，兼容的CUDA 版本以11.x 开头，cuDNN 版本以8.x 开头，Tensorflow 版本2.4 开头。

您可以尝试相应地升级 cuDNN 版本，您的问题将得到解决。更多详情可以参考CuDNN support matrix。