RuntimeError: CUDA error: CUBLAS STATUS ALLOC FAILED when calling "cublasCreate(handle)

这个错误通常是由于显存不足导致的。以下是两种解决方法：
1. 强制同步并观察错误代码行数：
在执行训练程序的时候将环境变量CUDA_LAUNCH_BLOCKING=1设为1(强制同步)可以准确定位观察到我们显存操作的错误代码行数。
2. 将代码在CPU上运行：
将代码在CPU上运行，不用GPU，这时候再报错就会给出具体出现问题的代码位置，有针对性的进行修改就好了。