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

问题：

最近在使用服务器跑分类模型时，出现了RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling cublasCreate(handle)的错误 。

解决方案：

在报错提示中，可以看到最后一行提示 t >= 0 && t < n_classes failed

/pytorch/aten/src/THCUNN/ClassNLLCriterion.cu:108: cunn_ClassNLLCriterion_updateOutput_kernel: block: [0,0,0], thread: [2,0,0] Assertion t >= 0 && t < n_classes failed.

很明显是我设置的分类标签越界的问题，重新检查我的语料，发现语料中对应的标签是[1,2,3,4,5]，故根据提示修改为[0,1,2,3,4].

再次运行，报错解决！