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

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

你可能感兴趣的:(深度学习)