最新消息:

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

操作系统 电脑手机教程网 74浏览


    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、验证
    

转载请注明:电脑手机教程网 » window10下pytorch和torchvision CPU版本安装教程