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

[RyanYip-Kat]

When I run python train.py,I got the error : RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle). I had installed the packages via "pip install -r requirement“,and try to change the batch_size，this error still occured.And how can I fixed it.Thanks!

[nullgogo]

遇到了同样的问题，请问如何解决啊？

[hyb0017]

pytorch incompatibility with CUDA toolkit can cause this problem