python - PyTorch:copy_if 同步失败:触发设备端断言

得到以下错误:



尝试执行时:
if torch.nonzero(gt_class_ids > n_classes).size()[0] > 0:
其中 gt_class_ids 是大小为 [128] 且 Torch.cuda.LongTensor = 81 的 n_classes。
在 cuda 9.x 上运行

最佳答案

以下代码对我有用。

n_classes = 81
gt_class_ids = torch.from_numpy(numpy.random.randint(1, 100, size=128)).long()

if torch.nonzero(gt_class_ids > n_classes).size(0) > 0:
    print('okay')

一个建议:在不使用 cuda 的情况下运行代码，然后您将能够看到真正的错误消息。有时，当我们使用 cuda 运行代码时，它会给出错误消息，其中包含隐藏真实错误消息的 device-side assert triggered。

关于python - PyTorch:copy_if 同步失败:触发设备端断言，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/50852294/