【问题描述】
使用GPU训练模型(.fit)时报错,返回Blas GEMM launch failed。使用nvidia-smi查看GPU状态,发现显存几乎占满,而GPU使用率极低。
【问题解决】
使用nvidia-smi查看GPU状态,找出占显存最多的一个进程,记下其PID号,使用kill PID号杀掉进程,即可正常训练。
【补充说明】
(nvidia-smi显示的GPU状态)

本文介绍了一种在使用GPU进行模型训练时遇到BlasGEMMlaunchfailed错误的解决办法。通过nvidia-smi工具检查并找到占用大量显存的进程,记录其PID并终止该进程,从而解决训练过程中出现的问题。
【问题描述】
使用GPU训练模型(.fit)时报错,返回Blas GEMM launch failed。使用nvidia-smi查看GPU状态,发现显存几乎占满,而GPU使用率极低。
【问题解决】
使用nvidia-smi查看GPU状态,找出占显存最多的一个进程,记下其PID号,使用kill PID号杀掉进程,即可正常训练。
【补充说明】
(nvidia-smi显示的GPU状态)

3832
7819

被折叠的 条评论
为什么被折叠?
时报错:Blas GEMM launch failed&spm=1001.2101.3001.5002&articleId=115585985&d=1&t=3&u=cc8cf7bafe724c62970512fa0ed2208f)