【已解决】RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasLtMatmul

最近在用BERT训练文本分类，报错RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasLtMatmul，百度一下都是when calling cublasSgemm，解决方法都很麻烦还不一定有效，google一下发现stackoverflow上一位老哥提到是GPU的问题

果断将batch_size调小，成功解决。