已解决！！！RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

最新推荐文章于 2025-02-25 16:30:37 发布

叫我梨就好了

最新推荐文章于 2025-02-25 16:30:37 发布

阅读量1.6k

点赞数

文章标签：深度学习人工智能 pytorch python

本文链接：https://blog.youkuaiyun.com/qq_41973051/article/details/132178063

版权

训练时出现了这样的警告：

/pytorch/aten/src/THCUNN/SpatialClassNLLCriterion.cu:106: cunn_SpatialClassNLLCriterion_updateOutput_kernel: block: [1,0,0], thread: [259,0,0] Assertion `t >= 0 && t < n_classes` failed.
/pytorch/aten/src/THCUNN/SpatialClassNLLCriterion.cu:106: cunn_SpatialClassNLLCriterion_updateOutput_kernel: block: [1,0,0], thread: [260,0,0] Assertion `t >= 0 && t < n_classes` failed.
/pytorch/aten/src/THCUNN/SpatialClassNLLCriterion.cu:106: cunn_SpatialClassNLLCriterion_updateOutput_kernel: block: [1,0,0], thread: [261,0,0] Assertion `t >= 0 && t < n_classes` failed.
/pytorch/aten/src/THCUNN/SpatialClassNLLCriterion.cu:106: cunn_SpatialClassNLLCriterion_updateOutput_kernel: block: [1,0,0], thread: [262,0,0] Assertion `t >= 0 && t < n_classes` failed.
/pytorch/aten/src/THCUNN/SpatialClassNLLCriterion.cu:106: cunn_SpatialClassNLLCriterio

最低0.47元/天解锁文章