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
安装的过程中会卸载之前安装的版本