RuntimeError: lines.type().is_cuda()INTERNAL ASSERT FAILED at ““, please report a bug to PyTorch.

沉迷单车的追风少年

于 2022-12-19 15:52:37 发布

阅读量1.3k

点赞数 5

CC 4.0 BY-SA版权

分类专栏：经验问题汇总文章标签： bug pytorch 人工智能深度学习

本文链接：https://blog.csdn.net/qq_41895747/article/details/126887763

经验问题汇总专栏收录该内容

163 篇文章 ¥9.90 ¥99.00

订阅专栏

超级会员免费看

在复现sketch R2CNN代码并进行neuralline rasterize操作时，遇到RuntimeError。错误源于模型和向量不在同一设备上。通过将所有模型和张量移至GPU，成功解决问题。

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

项目场景：

复现sketch R2CNN代码，neuralline rasterize操作。

问题描述

完整报错：

Traceback (most recent call last):
  File "attention_map.py", line 120, in <module>
    output = model(x)
  File "/root/anaconda3/envs/torch1/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
    return forward_call(*input, **kwargs)
  File "attention_map.py", line 108, in forward
    images = RasterIntensityFunc.apply(points, intensities, self.img_size, self.thickness, self.eps, self.device)
  File "/root/sketchtext/sketch-diffusion-main/neuralline/rasterize.py", line 35, in forward
    line_weight_map, locks, img_size, thickness, eps)
RuntimeError: lines.type().is_cuda()INTERNAL ASSERT FAILED at "/root/sketchtext