question

[AFESDTTM]

How can I handle this essue.
File "/data/ss/MAST/MAST/models/loss_ops.py", line 270, in geodesic_dists
internal = 0.5 * (torch.diagonal(torch.matmul(R_bin_ctrs, torch.transpose(R,-1, -2)), dim1=-1, dim2=-2).sum(-1) - 1.0)
RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)

[aragakiyui611]

I didn't meet this problem before, try install pytorch==1.8.1