[解决]RuntimeError: CUDA error: device-side assert triggered

有很多博客说是因为分类数目错,...blabla错,都不对!

这个报错是因为在gpu上运行时无法判断具体程序运行到什么地方报错,也有可能由硬件引发的错误。建议将程序放在CPU上运行debug,device='cpu'。

我自己的问题是torch.gather()中的索引超出最大长度,这个问题只有在cpu上debug是才能准确报出。

参考链接

[已解決] RuntimeError: CUDA error: device-side assert triggered - Clay-Technology World

你可能感兴趣的:(pytorch,人工智能,python)