标签：sse   使用   网上   math   报错   pytorch   htm   解决   宽度   

Pytorch报错：cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMath.cu:26

这种问题是网上比较常见的，一般的原因就是Label没有从0开始导致数组或者tensor超出范围。我这次也是这个原因，具体来说，是由于使用了nll_loss造成的。关于NLLLoss，可以看我的这篇文章。

在计算NLLLoss时，要算X_label,但是输入的label是从1到10，而X的宽度就是10，所以宽度就超了，修改之后问题就解决了。

Pytorch报错：cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMath.cu:26

标签：sse   使用   网上   math   报错   pytorch   htm   解决   宽度   

原文地址：https://www.cnblogs.com/jiading/p/11979406.html