pytorch使用中的错误及解决措施(持续更新)

1.IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number

 原因:出现此问题的原因是版本升级.

解决措施:将loss.data[0] 改为loss.item()

2.ValueError: could not broadcast input array from shape (7,7) into shape (7,8)

问题描述:出现此问题是在训练加载图片时出现的维度不匹配,在网上查了好久发现网上的解决措施都不管用,最后发现是pytorch版本的原因,本人安装的是pytorch1.1的,但是这个代码应该在低版本下跑,于是又重新装了0.4版本的.在这里需要注意的是对于pytorch的安装,网上一般会推荐;两种方式,一种是基于conda命令的,一种是基于pip命令的,本人采用的是基于pip命令的

pip2 install --user torch==0.4.0 -f https://download.pytorch.org/whl/cu90/stable

要注意,一般网上的教程是随着pytorch一起安装torchvision,但是在卸载的时候不但要

pip2 uninstall torch

而且需要把torchvision一起卸载掉,否则会出现不匹配的问题:

在卸载掉之后,再重新安装合适的版本即可:

pip2 install --user torchvision==0.2.0