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

Question

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

程序似乎停在这一行：

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

在此处输入图像描述 大约需要> = 30分钟...

有人在这里说它是关于 GPU 内存的。

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

标签： tensorflowkeras

Answer 1

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

您可以尝试相应地升级 cuDNN 版本，您的问题将得到解决。有关更多详细信息，您可以参考CuDNN 支持矩阵。