RuntimeError: cuda runtime error(59) : device-side assert triggered で詰まった話

pytorchでMLP使って多クラス分類したときのエラーについて書きます。

環境

• windows 10
• PyTorch v1.2

エラー内容

RuntimeError: cuda runtime error(59) : device-side assert triggered

私のソースコードではbackward()の箇所でエラーが発生していました

解決法

多クラス分類の正解ラベルのtensorは[0 ... n-1] のように最小値が0になるように変更する

(正解ラベルリスト = [2, 3, 4, 2] を [0, 1, 2, 0] に書き換えてからTensorDatasetへ)

参考 : gpu - CUDA runtime error (59) : device-side assert triggered - Stack Overflow

---

正解ラベルって0から始めるのが一般的だからこんなエラーは起きるはずもなく...