PyTorch 正在使 jupyter 笔记本崩溃，同时使用 nn 模块计算损失？

Question

model = nn.Sequential(nn.Linear(784, 128),
                  nn.ReLU(),
                  nn.Linear(128, 64),
                  nn.ReLU(),
                  nn.Linear(64, 10))

criterion = nn.CrossEntropyLoss()

images, labels = next(iter(trainloader))
images = images.view(images.shape[0], -1) 
logits = model(images)
loss = criterion(logits, labels) 
print(loss)

代码在“loss = criteria(logits, labels)”行崩溃。
数据集：MNIST 手写数字。

Answer 1

就我而言， 我通过将 pytorch 版本从 v1.0 降级并安装到 0.4.0 解决了这个问题。
如何安装 v0.4.0 ？
只需将版本名称从 1.0.0 更改为 0.4.0

1. pip3 安装https://download.pytorch.org/whl/cpu/torch-1.0.0-cp36-cp36m-linux_x86_64.whl
2. pip3 安装 torchvision

到

1. pip3 安装https://download.pytorch.org/whl/cpu/torch-0.4.0-cp36-cp36m-linux_x86_64.whl
2. pip3 安装 torchvision

这适用于 Ubuntu 18.04 操作系统，使用 pip，非 CUDA 安装。