在 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 專業版
uj5u.com熱心網友回復：
從終端運行相同的腳本顯示不同的錯誤，表明檔案
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 檔案夾解決了這個問題。
轉載請註明出處，本文鏈接：https://www.uj5u.com/yidong/324605.html
上一篇：減少SSD-300的訓練步驟