python - GPU训练时Tensorflow 2.5退出代码-1073740791

Question

在 GPU 上训练 Tensorflow 模型时，调用：

model.fit(...)

立即以代码-1073740791 (0xC0000409)退出：

Epoch 1/500
2021-10-16 20:13:42.154951: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library cudnn64_8.dll

Process finished with exit code -1073740791 (0xC0000409)

在 CPU 上做同样的事情没有任何问题。

系统信息：

• Python版本：3.8
• CUDA 版本：11.2
• cuDNN 版本：8.1
• 张量流版本：2.5
• 操作系统：Windows 10 专业版

标签： pythontensorflowgpuexit-code

Answer 1

从终端运行相同的脚本会显示不同的错误，表明该文件

cudnn_adv_infer64_8.dll

在路径中找不到。

从 cuDNN bin 文件夹中移动其余文件，即：

cudnn_adv_infer64_8.dll
cudnn_adv_train64_8.dll
cudnn_cnn_infer64_8.dll
cudnn_cnn_train64_8.dll
cudnn_ops_infer64_8.dll
cudnn_ops_train64_8.dll

进入CUDA bin文件夹解决了这个问题。