最近在使用服务器跑分类模型时,出现了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]
.