unable to get repr for ＜class ‘torch.Tensor‘＞和RuntimeError: CUDA error: device-side assert trigger错误

问题描述：

我的模型在本地运行没有任何问题，但是在远程调试的时候出现unable to get repr for ＜class ‘torch.Tensor‘＞和RuntimeError: CUDA error: device-side assert triggered错误。单步调试之后，定位到下图perm2变量处：

perm2 = torch.randperm(negative_numel, device=negative.device)[:num_neg]  

得到的结果应该是128个随机索引，且都属于[0,160742]范围内。

但是这里不知道为什么最后输出的是如下128个索引：

 

 可以看到，不知道为什么随机索引直接跑飞了，如果有知道什么原因造成的可以私信或评论我一下。

解决方法：

去掉device=negative.device，变为：

positive_numel = positive.numel()
negative_numel = negative.nume