1. 選擇合適的安裝包下載
https://anaconda.org/pytorch/repo?type=conda&label=main
2. 安裝命令: conda install 安裝包
3. 安裝錯誤:
在安裝torchvision可能會出現以下錯誤
ImportError: /home/anaconda3/envs/maskrcnn-benchmark/lib/python3.6/site-packages/torchvision/_C.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN3c1019UndefinedTensorImpl10_singletonE
原因:torchvision的版本與pytorch的版本不兼容
解決:選擇高版本或者低版本的torchvision,如 pytorch-1.0.1與torchvision-0.2.0匹配成功
命令:pip install torchvision==0.2.2
安裝的過程中會卸載之前安裝的版本