RuntimeError: 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)`

我的问题出现的代码地方：

百度的时候看到这个问题的两个解决办法：

1.可能是shape不对，但是我查看了自己的shape，没有问题

2.可能是放在了不同的device上，我的也没有这个问题，都在cuda0上

 

 问题解决：

复现的代码需要在torch1.2下运行，所以我的环境是1.2，我换成了和cuda环境相匹配的torch1.8，就运行成功了