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

大多数问题是说显存爆了，如果不是显存爆了。我在做图像分类的时候，代码配置里需要手动配置图像的类别，如果类别数和你的数据集的类别数不一致，我就报了这个错。

尝试检查类别数目是否设置正确。