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

Tensorflow 2.5 exit code -1073740791 when GPU training(显卡培训时TensorFlow2.5退出代码-1073740791)

本文介绍了显卡培训时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文件夹解决了此问题。

这篇关于显卡培训时TensorFlow2.5退出代码-1073740791的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持编程学习网!

本文标题为:显卡培训时TensorFlow2.5退出代码-1073740791

基础教程推荐