原文: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