RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasGemmEx( handle, opa, opb, m, n, k, &falpha, a, CUDA_R_16F, lda, b, CUDA_R_16F, ldb, &fbeta, c, CUDA_R_16F, ldc, CUDA_R_32F, CUBLAS_GEMM_DFALT_TENSOR_OP)
现在报的错,越来越找不到解决方法了,各位有解决方法吗?
我后面改小了batch_size并调小了图像输入大小,大概就是cuda的计算能力不够了
博主在训练模型时遇到了CUDA错误CUBLAS_STATUS_EXECUTION_FAILED,尝试通过减小batch_size和图像输入大小来缓解计算资源不足的问题,但问题依然存在。寻求社区帮助寻找更有效的解决方案。
8万+

被折叠的 条评论
为什么被折叠?



