解决方案:
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)`
在进行上述解决方案后,也没有报这个错误,两者应该是同一个原因造成的
解决方案:
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)`
在进行上述解决方案后,也没有报这个错误,两者应该是同一个原因造成的