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

这个报错出现在计算loss的时候，代码原先是正常的。

我尝试了很多方法，我把cuda换成在cpu上跑，就正常运行了。

如果问题不是出在这，你可尝试在cpu上跑一下，看看具体跑什么错。

原因是显存不够造成的，可以把batch改小或者imagesize改小。