Image 模型在丢失时停止；“非法指令提示”；

类似于但OP说他的程序在linux上成功运行，而我的问题是在Ubuntu上。我试过降低torch和torchvision的等级，但问题依然存在。我怀疑导致问题的部分如下所示：

for epoch in range(max_epoch):

    running_loss = 0.0
    for image, label in train_loader:

        optimiser.zero_grad()

        output = model(image)

        print("Label:{}\n".format(label))
        print("Output is {}\n".format(output))

        loss = criterion(output,label)
        print("Loss is {}".format(loss.item()))
        loss.backward()
        print("Backward loss and now optimiser")
        optimiser.step()

        running_loss += loss.item()
        print("Loss: {:.4f}".format(running_loss))

完整代码可用 这是我得到的提示：

---

有一个帖子在谈论这个问题，谢谢你。似乎这里几乎没有人能做到什么是SO、环境和pytorch版本？Ubuntu18.04、pip、torch 1.4.0 cpu和torchvision 0.5.0 cpu。这是torch的最新版本，加上我也使用了前面的版本。这里基本相同。您能否创建一个虚拟环境并尝试在那里复制此错误？另外，你能把所有东西都移动到GPU上看看会发生什么吗？