../aten/src/ATen/native/cuda/IndexKernel.cu:92: operator(): block: [651,0,0], thread: [127,0,0] Asse

迈不过友情╰ 2024-02-17 10:14 21阅读 0赞

#### RuntimeError: CUDA error: device-side assert triggered ####

*  问题
 *  原因
 *  解决方案

## 问题 ##

![在这里插入图片描述][361122526fea4285b520b07e6fae5a23.jpeg_pic_center]

## 原因 ##

Assertion \`index >= -sizes\[i\] && index < sizes\[i\] && “index out of bounds”

数据超出了边界。在给出label的时候，我的数据集在某个标签上会给进去一个小于零或者大于类别数的一个标签，大白话就是设定了模型分类数量为4，但是数据集中有大于4个类别则会报错，少于4个是不会报错。

解决方法：修改预设的类别数量，class\_map里面预先写好的类别个数和实际的数据集里面的标签种类和个数不同。可以要修改class\_map或者修改数据集。

![在这里插入图片描述][9e41085ebc5c475aa397b41148e8a276.png]  
这里有96的类别  
但是，代码中最高95.。。

![在这里插入图片描述][4bffc75d2cf4426a82727b692d938c86.jpeg_pic_center]  
是之前02标签转换的时候类别错了。。。  
[参考1][1]

[一样的问题][Link 1]

## 解决方案 ##

删掉02的图像和标签

rm -r 00000061*

显示文件夹下jpg图像数量

ls -lR | grep ".*jpg" | wc -l

![在这里插入图片描述][6f9e73ce753a4db6bb52496a3720a8a9.jpeg_pic_center]

[Linux上统计文件夹下文件个数以及目录个数][Linux]

[数据处理过程][Link 2]

实时查看nvidia显卡显存

watch -n 1 nvidia-smi

\#其中，1表示每隔1（可改）秒刷新一次终端的显示结果

[361122526fea4285b520b07e6fae5a23.jpeg_pic_center]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/06/2b61c8d57cf140ddabd7fa2cf96e0d20.png
[9e41085ebc5c475aa397b41148e8a276.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/06/6fa0da47e0134bee8ef09a086b838e89.png
[4bffc75d2cf4426a82727b692d938c86.jpeg_pic_center]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/06/6c9ce4dcddb54d65bfb5b3dfcf63870c.png
[1]: https://blog.csdn.net/qq_41900846/article/details/128536185
[Link 1]: https://bbs.csdn.net/topics/616640970
[6f9e73ce753a4db6bb52496a3720a8a9.jpeg_pic_center]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/06/559edb5ca0204233ba3f93c9a8ba5185.png
[Linux]: https://www.cnblogs.com/yongjieShi/p/8075281.html
[Link 2]: https://blog.csdn.net/xu1129005165/article/details/132666266