您的位置:首页 > 其它

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

2017-12-02 10:06 766 查看
使用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
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐