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

Question

出现以下错误:

RuntimeError: copy_if failed to synchronize: device-side assert triggered

尝试执行时:

如果 torch.nonzero(gt_class_ids > n_classes).size()[0] > 0:

其中 gt_class_ids 是大小为 [128] 且 n_classes = 81 的 Torch.cuda.LongTensor。
在 cuda 9.x 上运行

Answer 1

以下代码适合我。

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 运行代码时，它会给出错误消息 having device-side assert triggered 这隐藏了真正的错误消息。