CUDA-error: device-side triggered的问题所有的报错可能性

第一种就是你模型用了分类,然后分类的输出和标签的数量对不上。就比如说输出的是0-55,标签是1-56.这样会报错。
第二种就是之前加了torch.cuda.empty_cache(),有可能会报错。虽然还不完全清楚为什么会这样,但是对我来说是这样的。

你可能感兴趣的:(python)