在 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 专业版
从终端运行相同的脚本显示不同的错误,表明该文件
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 文件夹解决了该问题。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)