【GPU报错解答】Internal: Blas SGEMM launch failed

原创于 2021-03-23 21:49:21 发布 · 1.3k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow

算法小白进阶之路专栏收录该内容

7 篇文章

订阅专栏

文章讲述了作者遇到的GPU内部错误BlasSGEMMlaunchfailed，尝试了多种常见解决方案如重启kernel和更新cuDNN无效。最终，通过在代码中设置GPU内存增长策略成功解决了问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

文章目录

1.报错
2.思路＆解决方案

1.报错

Internal: Blas SGEMM launch failed

2.思路＆解决方案

思路：
查了各种博客是GPU和cuDNN的问题
大部分解决方案是重启kernel，重新安装cuDNN，安装cuDNN补丁等等
但是对我来说，都没用

最终解决方案：
在运行文件上面加上这一段

#防止cuDNN报错：方案2
gpus = tf.config.experimental.list_physical_devices('GPU')
if gpus:
    try:
        # Currently, memory growth needs to be the same across GPUs
        for gpu in gpus:
            tf.config.experimental.set_memory_growth(gpu, True)
        logical_gpus = tf.config.experimental.list_logical_devices('GPU')
        print(len(gpus), "Physical GPUs,", len(logical_gpus), "Logical GPUs")
    except RuntimeError as e:
        # Memory growth must be set before GPUs have been initialized
        print(e)