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

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐