【tensorflow报错】tensorflow.python.framework.errors_impl.InternalError: Blas GEMM launch failed：XXX

最新推荐文章于 2024-09-26 15:20:13 发布

人工智能程序源

最新推荐文章于 2024-09-26 15:20:13 发布

阅读量9.5k

点赞数 8

分类专栏： Tensorflow 文章标签： tensorflow python 深度学习

本文链接：https://blog.csdn.net/weixin_44445154/article/details/121661639

版权

在使用tf-gpu=2.3、cuda=10.1、cudnn=7.4.2环境中遇到Blas GEMM launch failed错误。尝试了调整代码、管理GPU占用、清理缓存、修改batch_size等方法无效，最终通过重装NVIDIA驱动、cuda和cudnn解决问题。下篇将分享详细解决步骤。

摘要由CSDN通过智能技术生成

背景：

最近遇到了一个较为头痛的问题，连续改BUG两天最终才解决掉，解决此问题流程及方法如下：
此问题是在tf-gpu=2.3、cuda=10.1、cudnn=7.4.2环境下出现在（tf的cpu版本时，是可以跑通模型的）。

具体BUG情况如下：

1、failed to create cublas handle: CUBLAS_STATUS_NOT_INITIALIZED

2、Blas GEMM launch failed :

tensorflow.python.framework.errors_impl.InternalError: Blas GEMM launch failed : a.shape=(2048, 2), b.shape=(2, 768), m=2048, n=768, k=2

[[{ {nodeStatefulPartitionedCall/StatefulPartitionedCall/transformer_encoder/StatefulPartitionedCall/type_embeddings/MatMul}}]] [Op:__forward_restored_function_body_32773]

Function call stack:
restored_function_body