• 分类

  • 重置

window10下pytorch和torchvision CPU版本安装教程

    1、环境

    python3.5

    Anaconda 4.2.0

    2、pytorch安装

    
    pip3 install https://download.pytorch.org/whl/cpu/torch-1.1.0-cp35-cp35m-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/

    3、torchvision安装

    
    pip3 install https://download.pytorch.org/whl/cpu/torchvision-0.3.0-cp35-cp35m-win_amd64.whl

    注意:直接pip3 install torchvision ,即使成功安装了,但是在导入的时候import torchvision会报错:

    网上解决办法:(1)降低Pillow版本,即Pillow==4.1.1;(2)conda install -c peterjc123 vc vs2017_runtime

    两个办法都试了,但都无效,不知是否因为跑pytorch是cpu版本缘故,但是从这个链接进行安装就可以成功:

    
    https://download.pytorch.org/whl/cpu/torchvision-0.3.0-cp35-cp35m-win_amd64.whl

    4、验证

    以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持lingkb。