RuntimeError: CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling cublasCreate(handle)

参考https://github.com/lucidrains/big-sleep/issues/48，将torch从1.8.0更新至1.8.1后问题得以解决。