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

问题:

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

RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublasCreate(handle)`_第1张图片

解决方案:

在报错提示中,可以看到最后一行提示 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].

RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublasCreate(handle)`_第2张图片

再次运行,报错解决!

你可能感兴趣的:(报错解决,python,pytorch)