当前位置:主页 > 服务器 > Windows >

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

时间:2020-10-07 14:28:54 | 栏目:Windows | 点击:

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

您可能感兴趣的文章:

相关文章