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)

Сбой кода в строке " убыток = критерий (логиты, метки) ".
Набор данных : рукописные цифры MNIST.

Answer 1

В моем случае я решил эту проблему путем понижения и установки версии pytorch до версии 0.4.0 с версии 1.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 установка.