PyTorch：copy_if无法同步：触发设备端断言

Question

收到以下错误：

  

RuntimeError：copy_if无法同步：触发设备端断言

尝试执行时：

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

其中gt_class_ids的大小为[{1] Torch.cuda.LongTensor，n_classes = 81。 在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运行代码时，它会给出device-side assert triggered的错误消息，该消息会隐藏真实的错误消息。