【解决bug】RuntimeError: cuda runtime error (77) : an illegal memory access was encountered at XXXXX

最新推荐文章于 2025-11-10 10:06:17 发布

原创

最新推荐文章于 2025-11-10 10:06:17 发布 · 2.2k 阅读

2 ·

CC 4.0 BY-SA版权

在虚拟试穿项目训练中，遇到CUDA运行时错误77，源于label_map数据问题。通过调整label_tensor为transform_A(label_data) * 255.0，解决了内存非法访问。确保运算tensor数量级一致避免越界。

项目场景：

虚拟试穿项目训练：
项目场景：数据集准备完成，构建模型开始训练时，模型内部编码数据时:
input_label, masked_label, all_clothes_label = self.encode_input(label, clothes_mask, all_clothes_label)报错：
RuntimeError: cuda runtime error (77) : an illegal memory access was encountered at XXXXXX

    def encode_input(self, label_map, clothes_mask, all_clothes_label):
        size = label_map.size()
        oneHot_size = (size[0], 14, size[2], size[3