pytorch中常见错误总结

错误1:在console中输入import torchvision会报错：ImportError: cannot import name 'PILLOW_VERSION'

原因：torchvision和Pillow版本不兼容，可能pillow版本过高，如pillow 7.0和torchvision 0.3.0冲突。

解决办法：降低Pillow版本，即输入：

pip install Pillow==6.2.2

错误2：IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python

原因：train_loss += loss.data[0] 是pytorch0.3.1版本代码,在0.4-0.5版本的pytorch会出现警告,不会报错,但是0.5版本以上的pytorch就会报错,总的来说是版本更新问题.

解决办法：

loss.data[0]修改为loss.item()