天天看點

RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasSgemm( handle, opa, opb, m

RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling

cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)

錯誤的原因是nn.Linear() 次元錯了