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

 更新时间:2019年10月17日 11:17:57   作者:当太阳不再发光   我要评论
这篇文章主要介绍了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、验证

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

相关文章

最新评论