跑深度学习模型报错:Function call stack:train_function -> train_function

部署运行你感兴趣的模型镜像

今天在跑自己设计的深度学习模型时出现了以下问题:

InternalError: 2 root error(s) found.
  (0) Internal:  Blas GEMM launch failed : a.shape=(12, 3), b.shape=(3, 32), m=12, n=32, k=3
	 [[node model/dense_3/MatMul (defined at <ipython-input-16-5a73e5716594>:10) ]]
	 [[gradient_tape/model/token_and_position_embedding/embedding_1/embedding_lookup/Reshape/_98]]
  (1) Internal:  Blas GEMM launch failed : a.shape=(12, 3), b.shape=(3, 32), m=12, n=32, k=3
	 [[node model/dense_3/MatMul (defined at <ipython-input-16-5a73e5716594>:10) ]]
0 successful operations.
0 derived errors ignored. [Op:__inference_train_function_3032]

Function call stack:
train_function -> train_function

出现该问题的原因可能是各种库之间的版本依赖问题,或者是GPU被占用了。

而我的情况是GPU被占用了,因此我把其他占用GPU的程度都关了,再次跑模型,OK了!

您可能感兴趣的与本文相关的镜像

Wan2.2-I2V-A14B

Wan2.2-I2V-A14B

图生视频
Wan2.2

Wan2.2是由通义万相开源高效文本到视频生成模型,是有​50亿参数的轻量级视频生成模型,专为快速内容创作优化。支持480P视频生成,具备优秀的时序连贯性和运动推理能力

评论 1
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值