RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasGemmEx报错

解决方案:

unset LD_LIBRARY_PATH

之前还出现错误:

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

在进行上述解决方案后,也没有报这个错误,两者应该是同一个原因造成的

你可能感兴趣的:(linux)