pytorch运行报错，原因，原始标签和pytorch的标签数字不匹配

PyCharm报错：RuntimeError: CUDA error: device-side assert triggered 问题的解决思路

1、问题描述

RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemmStridedBatched( handle, opa, opb, m, n, k, &alpha, a, lda, stridea, b, ldb, strideb, &beta, c, ldc, stridec, num_batches)

最近在做多分类过程中遇到了这个问题，看了许多文章，报错的原因大概就是，我的标签是1-17，而在pytorch中，它使用的是0-16的label，当数据中出现标签17，就没有办法识别。

2、解决办法

将标签数值整体-1，实现和pytorch 标签统一的效果。

return data, label-1

参考资料：
链接: https://blog.csdn.net/Penta_Kill_5/article/details/118085718