【PyTorch】RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling cublasSgemm()

完整报错信息.RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE whencalling `cublasSgemm( handle, opa, opb, m, n, k, &alpha, a,lda, b, ldb, &beta, c, ldc)`出错原因与解决方案.通常为线性层 n n . L i n e a r \rm n

qq_43208491

8283人浏览 · 2021-09-19 17:17:00

qq_43208491 · 2021-09-19 17:17:00 发布

完整报错信息.

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

出错原因与解决方案.
通常为线性层 n n . L i n e a r \rm nn.Linear nn.Linear 定义时数据维度参数与实际数据维度不匹配，需要检查核对后修改。
原文链接：https://blog.csdn.net/weixin_44246009/article/details/119426147