原文:GPU 版 TensorFlow failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED

原因: 使用 GPU 版 TensorFlow ,並且在顯卡高占用率的情況下 比如玩游戲 訓練模型,要注意在初始化 Session 的時候為其分配固定數量的顯存,否則可能會在開始訓練的時候直接報錯退出。 解決方法: 原先代碼: 現在代碼: ...

2019-08-01 11:15 0 1195 推薦指數:

查看詳情

 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM