import torch
import torchvision
报错信息如下:在import torchvision这行就出错了!
raise NotSupportedError(base.range(), "slicing multiple dimensions at the same time isn't supported yet")
torch.jit.frontend.NotSupportedError: slicing multiple dimensions at the same time isn't supported yet
proposals (Tensor): boxes to be encoded
"""
# perform some unpacking to make it JIT-fusion friendly
wx = weights[0]
wy = weights[1]
ww = weights[2]
wh = weights[3]
proposals_x1 = proposals[:, 0].unsqueeze(1)
~~~~~~~~~ <--- HERE
proposals_y1 = proposals[:, 1].unsqueeze(1)
proposals_x2 = proposals[:, 2].unsqueeze(1)
proposals_y2 = proposals[:, 3].unsqueeze(1)
reference_boxes_x1 = reference_boxes[:, 0].unsqueeze(1)
reference_boxes_y1 = reference_boxes[:, 1].unsqueeze(1)
reference_boxes_x2 = reference_boxes[:, 2].unsqueeze(1)
reference_boxes_y2 = reference_boxes[:, 3].unsqueeze(1)
网上查看了一下为什么出错了,看见大多数都说是torch和torchvision的版本不匹配问题。然后在anaconda3里面命令行输入:pip list查看所有已经安装的包和版本信息,如下图:(我感觉这个是对的呀?)好吧,继续百度发现了好像确实版本不对——旧版本的0.4.1的torch对应的torchvision是0.2.2的。注:cu表示的是cuda的版本,cp表示的是python的版本,后面-i是下载的清华镜像
先卸载原来的torchvision:
再命令行查看是否成功卸载:
现在开始重新安装:(由于是旧版本,所以选择的离线安装更快。参照上一篇windows10+python3.6+cuda) 好尴尬!没有对应的cu92 cp36的torchvision0.2.2版本,哪里找呢?https://download.pytorch.org/whl/torch_stable.html(这个也没有!)
看见也有人说下载好了torch之后,anaconda会根据pytorch的版本来下载相应版本的torchvision,那么试试吧!
conda install torchvision
我就知道不行!最开始配置pytorch的时候就试过这个方法了!我太难了!
功夫不负有心人!直接命令行:pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple/ torchvision==0.2.2