2019年8月1日
GPU 版 TensorFlow failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
摘要： 原因： 使用 GPU 版 TensorFlow ，并且在显卡高占用率的情况下（比如玩游戏）训练模型，要注意在初始化 Session 的时候为其分配固定数量的显存，否则可能会在开始训练的时候直接报错退出。 解决方法： 原先代码： 现在代码： 阅读全文
posted @ 2019-08-01 11:15 巽逸 阅读(2719) 评论(0) 推荐(0) 编辑