【PyTorch报错已解决】CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm( handle, opa, opb

报错

  • 报错信息为:发生异常: RuntimeError CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)
    在这里插入图片描述

  • 尝试了但都无效的方法:

    • 在main函数中添加CUDA_LAUNCH_BLOCKING=1
    • 重启环境;
    • 尝试更小的矩阵乘法(排除了显存不够或者内存不够的问题);
    • 替换 @ 、 torch.mm、torch.matmul、torch.einsum等不同的矩阵乘法函数。

有效解决方法

  • 最后的解决方法是:将张量计算转移到cpu上,并进行调试。定位到真实的问题。
	device = torch.device('cuda:1' 
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值