RuntimeError: invalid argument 4: out of range at pytorch/torch/lib/TH/generic/THTensor.c:439

使用from torchvision.utils import save_image保存图像时,出现错误:

File “/usr/local/lib/python2.7/dist-packages/torchvision/utils.py”, line 51, in save_image

grid = make_grid(tensor, nrow=nrow, padding=padding)

File “/usr/local/lib/python2.7/dist-packages/torchvision/utils.py”, line 38, in make_grid
.narrow(2, x*width+1+padding//2, width-padding)\
RuntimeError: invalid argument 4: out of range at /pytorch/torch/lib/TH/generic/THTensor.c:439

原因是torchvision版本不对,解决方法是卸载torchvision,并重新安装,

pip uninstall torchvision 

pip install torchvision