会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
yjy888
博客园
首页
新随笔
联系
订阅
管理
2019年6月4日
keras 或 tensorflow 调用GPU报错:Blas GEMM launch failed
摘要: 我之前用keras 调用 tf 后端训练的时候,经常出现显存分配错误的问题。保险的方法是手动指定显存分配。 import tensorflow as tffrom keras.backend.tensorflow_backend import set_sessionconfig = tf.Confi
阅读全文
posted @ 2019-06-04 10:16 yjy888
阅读(1964)
评论(0)
推荐(0)
编辑
公告