1. 程式人生 > >解決辦法:KeyError: ‘ExpandBackward’及老版本pytorch/torchvision的安裝辦法。

解決辦法:KeyError: ‘ExpandBackward’及老版本pytorch/torchvision的安裝辦法。

  在做pytorch向caffe轉換的時候,報錯:

KeyError: ‘ExpandBackward’

  解決辦法:

  • 先解除安裝torch/torchvision。
  • 安裝老版本的pytorch
# 一般不成功。
# sudo pip2 install pytorch=0.2.0 

sudo pip2 install http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp27-cp27mu-manylinux1_x86_64.whl
  •  安裝老版本的torchvision
sudo pip2 install torchvision=0.1.8