pytorch计算错误unspecified launch failure

pytorch计算出现CUDA 错误unspecified launch failure或者CUBLAS_STATUS_INTERNAL_ERROR when calling `cublasSgemm

Pytorch进行深度学习计算CUDA错误

你好! 使用pytorch计算一直出现unspecified launch failure,并不是一开始计算就出现,而是随机出现这个错误,有时能一直计算,而有时候一开始就不能,通过总结发现,一般网络节点数一多,就很容易出现这个问题。
1、刚开始各种重装CUDAtoolkit与torch还有torchvision,没用。
2、更新pytorch等到最新版本,发现还是没用,出现新的错误:CUBLAS_STATUS_INTERNAL_ERROR when calling `cublasSgemm
然而还是不知道怎么解决这个问题。

解决办法

折腾了好久,一直不知道怎么办,使用了包括网上好多人所说的更改Nsight Monitor中的True到False,还是没用。
甚至换了显卡都没用,最后打算放弃的时候,两台机子一样的,但是一台可以算,一台不可以,突然发现了一个差异:显卡驱动不同,发现新版本的显卡驱动才会出现这个问题。
右键进入NVIDIA 控制面板:
在这里插入图片描述
看上面的版本,现在新机子版本比较高才不能计算。亲测版本431.60与431.70均可以。下面给上显卡各版本的驱动下载地址:https://www.geforce.cn/drivers/beta-legacy

  • 6
    点赞
  • 20
    收藏
    觉得还不错? 一键收藏
  • 15
    评论
评论 15
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值