完美解决:RuntimeError: CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling `cublasCreate(handle)

报错场景:

self.scores = nn.ParameterList()
        self.scores.append(Parameter(torch.FloatTensor(nfeat, 1)))
        for i in range(nhidlayer):
            self.scores.append(Parameter(torch.FloatTensor(nhid, 1)))

## 这一句报错,也就是做矩阵乘法@时报错
s_i = torch.sigmoid(fea @ self.scores[0] + self.bias[0])

我在网上搜了很多,都是啥embedding的原因,明显跟我身处的场景不一样

https://blog.csdn.net/xiaoxiaowantong/article/details/109689146

之后,我再stackflow找到了答案,https://stackoverflow.com/questions/66567324/pytorch-error-cuda-error-cublas-status-internal-error-when-calling-cublascrea

就是pytorch的版本有问题,我之前用的版本是1.8.0,按照解决办法,换成1.7.1的。

pip uninstall torch

pip install torch==1.7.1

  • 9
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 9
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值