Pytorch报错：CUDA error: device-side assert triggered

最新推荐文章于 2025-05-08 14:24:51 发布

努力写代码的小菜鸟

最新推荐文章于 2025-05-08 14:24:51 发布

阅读量4.7k

点赞数 2

CC 4.0 BY-SA版权

分类专栏：遇到的问题 Pytorch

本文链接：https://blog.csdn.net/weixin_37176244/article/details/116940011

在使用CUDA进行训练时遇到CUDA error: device-side assert triggered的报错，通过切换到CPU训练，发现错误源于损失函数的要求。解决方案是在比较时，确保数值在特定范围内，并添加适当条件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

总是遇到如下的报错，之前一直也没有记录解决的过程，今天记录一下，避免以后再踩坑

pytorch/aten/src/ATen/native/cuda/Loss.cu:106: operator(): block: [0,0,0], thread: [0,0,0] Assertion `input_val >= zero && input_val <= one` failed.
/pytorch/aten/src/ATen/native/cuda/Loss.cu:106: operator(): block: [0,0,0], thread: [1,0,0] Assertion `input_val >= zero && input_val <= one` failed.
/pytorch/aten/src/ATen/native/cuda/Loss.cu:106: operator(): block: [0,0,0], thread: [2,0,0] Assertion `input_val >= zero && input_val <= one` failed.
/pytorch/aten/src/ATen/native/cuda/Loss.cu:106: operator(): block: [0,0,0], thread: [3,0,0] Assertion `input_val >= zero && input_val <= one` failed.
/pytorch/aten/src/ATen/native/cuda/Loss.cu:106: operator(): block: [0,0,0], thread: [4,0,0] Assertion `input_val >= zero && input_val <= one` failed.
/pytorch/aten/src/ATen/native/cuda/Loss.cu:106: operator(): block: [0,0,0], thread: [5,0,0] Assertion `input_val >= zer