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 = standard（logits，标签）”行崩溃。
数据集：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操作系统，使用非CUDA安装的pip。