RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`

[Jiazhi_Li]

Thank you very much. When I update the latest version, the problem is solved.