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

如果你试过了网上查了很多错误方案依然无法解决，可以尝试以下操作。
这个错误是cuda抛出来的, 由于cuda的封装问题，很多错误都能导致这个错误抛出，更具体的错误会被隐藏在这个错误之下。
这个时候应该把device从cuda改成cpu，抛出的错误信息会更详细。