显卡培训时TensorFlow2.5退出代码-1073740791

问题描述

在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
• TensorFlow版本：2.5
• 操作系统：Windows 10专业版

---

解决方案

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

cudnn_adv_infer64_8.dll

在路径中找不到。

将文件的睡觉从cuDNNbin文件夹中移出，即：

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文件夹解决了此问题。