解决错误：RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`

在运行一个二分类的代码时候出现以下错误：

解决方法：

 1.减小batch_size。检查GPU内存，这个错误很可能是由于GPU的运行内存不够引起的。

2.更新GPU驱动程序。

3.重启

4.检查硬件问题

5.上述方案都不能解决时候（我的问题就是这样...），尝试检查一下输出和标签的匹配问题。如标签的数量不等于网络输出通道的数量，即预测的输出类的数量。调整输出以匹配，它应该可以解决问题。

此外，可以使用如下命令格式来输出它的底层问题

CUDA_LAUNCH_BLOCKING=1 python [你的命令参数]
eg.  CUDA_LAUNCH_BLOCKING=1 python train.py --task davis