pytorch出现:CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling `cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)

使用pytorch进行网络训练发生如下的错误,但是也不是所有的都会发生,感觉像是随机出现这个问题,可能一开始能计算,算几步之后就出现这个问题。cudatoolkit与pytorch各种重装,都没用,最近全部升级到最新版本,出现这个error
我的显卡是:RTX2060(笔记本电脑上的)
cudatoolkit:10.2.89
pytorch版本:1.6
麻烦有没有大神看看哪里有问题,实在是太烦人了。
图片说明

图片说明

请问楼主解决了吗?我也是这个问题呢

 

解决办法:https://blog.csdn.net/hunterflyy/article/details/108369274

https://blog.csdn.net/u014176855/article/details/80994991

该问题已解决,见此链接