离线下载pytorch安装包

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

安装的过程中会卸载之前安装的版本

    原文作者:pytorch
    原文地址: https://www.cnblogs.com/haiyang21/p/11216025.html
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞