原文:首次运行tensorflow-gpu 1.0 报错 failed to create cublas handle: CUBLAS_STATUS_NOT_INITIALIZED

发现博客: https: blog.csdn.net u article details 于是找到解决方法。 sudo rm rf .nv ...

2018-09-25 10:12 0 3426 推荐指数:

查看详情

GPUTensorFlow failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED

原因: 使用 GPUTensorFlow ,并且在显卡高占用率的情况下(比如玩游戏)训练模型,要注意在初始化 Session 的时候为其分配固定数量的显存,否则可能会在开始训练的时候直接报错退出。 解决方法: 原先代码: 现在代码: ...

Thu Aug 01 19:15:00 CST 2019 0 1195
 
粤ICP备18138465号  © 2018-2026 CODEPRJ.COM