InternalError (see above for traceback): Blas GEMM launch failed
分类:
TensorFlow
训练BiLSTM模型的时候报错:
1 2 3 | InternalError (see above for traceback): Blas GEMM launch failed : a.shape = ( 32 , 200 ), b.shape = ( 200 , 400 ), m = 32 , n = 400 , k = 200 [[node bilstm_crf / bidirectional_rnn / fw / fw / while / lstm_cell / MatMul (defined at train - nxf.py: 48 ) ]] [[node bilstm_crf / rnn_2 / while / Switch_2 (defined at train - nxf.py: 69 ) ]] |
解决方法:
如果你是使用 GPU 版 TensorFlow 的话,并且你想在显卡高占用率的情况下训练模型,那你要注意在初始化 Session 的时候为其分配固定数量的显存,否则可能会在开始训练的时候直接报错退出:
这时你需要用下面的方法创建 Session:
1 2 | gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction = 0.333 ) sess = tf.Session(config = tf.ConfigProto(gpu_options = gpu_options)) |
参考文献:
【1】https://blog.csdn.net/orangefly0214/article/details/80978374
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 使用C#创建一个MCP客户端
· ollama系列1:轻松3步本地部署deepseek,普通电脑可用
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· 按钮权限的设计及实现