bug全称
- RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE 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)
背景描述
ChatGLM3再进行模型微调的时候,启动报错,环境是cuda 12.0,torch2.0.1。
解决方案
可以通过conda创建虚拟环境,调整版本cuda 11.8,torch 2.1.0。正常启动,问题解决。