博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
windows上配置pytorch
阅读量:5227 次
发布时间:2019-06-14

本文共 1538 字,大约阅读时间需要 5 分钟。

操作系统:win10

已安装程序:Python 3.6 + Anaconda 5.1.0 + CUDA 9

pytorch官网:

1、进入官网,从Get Started进入pytorch安装教程,选择对应操作系统、python版本、安装方式、CUDA版本:

可选择多种安装方式,在Run this Command可看到对应命令:

1、Conda(Anaconda)

Command:

conda install pytorch torchvision -c pytorch

2、pip

Command:

pip3 install https://download.pytorch.org/whl/cu90/torch-1.0.0-cp36-cp36m-win_amd64.whlpip3 install torchvision //pip3不能用换成pip

3、Source

根据官网提示安装,网址:,个人认为没有前两种方便,毕竟cmd下可直接安装。

第一次我使用conda,在cmd命令行下安装,输入conda install pytorch torchvision -c pytorch,报错:找不到对应的包,就转为使用pip方法,后来查了一下,可以手动找包,参考Blog: 和 。(此种方法未经尝试,仅供参考)

接下来说一说pip方法安装:

进入anaconda prompt,运行command:pip install https://download.pytorch.org/whl/cu90/torch-1.0.0-cp36-cp36m-win_amd64.whl

下载后提示:exist pip 10, but pip 18.1 already exist,建议执行命令:python -m pip install --upgrade pip,应该是pip版本问题,所以我按建议更新了pip,此过程卸载了pip 10,安装了pip 18.1。更新完pip后,重新执行命令,即提示安装成功。再继续运行第二条命令,成功。

 

检查安装是否成功:

cmd下进入python,依次输入import torch和import torchvision,没报错即安装成功

在这里可以看到import torch安装成功,但torchvision安装失败。这里我再次pip install torchvision,出现以下提示:

 提示是已经成功安装的。再次输入import torchvision,还是提示相同的错误。查找之后发现,torchvision是有依赖环境的,就是上图的这几条:

而问题就出在第一条Pillow上:

在这里只要pillow版本>=4.1.1就可以,括号里标明了现在的版本是5.0.0,借鉴了有一个博主把pillow版本降到4.1.1的方法 ,执行pip install pillow==4.1.1:

把pillow版本降到4.1.1之后,真的成功了:

结论就是新版本的pillow缺少支持torchvision的东西,降版本可以解决这个问题。

 

 

其实可以直接进入cmd,使用pip方法安装,但我在cmd命令行下执行命令时,出现报错:'pip'不是内部或外部命令,也不是可运行的程序。我重新将pip所在的文件路径加入系统变量PATH里(python路径下的lib文件夹里:),还是找不到命令,于是转为anaconda prompt里执行。在更新了pip之后,进入cmd,输入pip -v检查pip命令是否可用:

pip命令更新后就可在cmd里正常使用。

 

转载于:https://www.cnblogs.com/zhenggege/p/10289867.html

你可能感兴趣的文章
QT文件读写
查看>>
C语言小项目-火车票订票系统
查看>>
[Linux]PHP-FPM与NGINX的两种通讯方式
查看>>
Java实现二分查找
查看>>
优秀员工一定要升职吗
查看>>
[LintCode] 462 Total Occurrence of Target
查看>>
springboot---redis缓存的使用
查看>>
架构图-模型
查看>>
sql常见面试题
查看>>
jQuery总结第一天
查看>>
Java -- Swing 组件使用
查看>>
Software--Architecture--DesignPattern IoC, Factory Method, Source Locator
查看>>
poj1936---subsequence(判断子串)
查看>>
黑马程序员_Java基础枚举类型
查看>>
[ python ] 练习作业 - 2
查看>>
一位90后程序员的自述:如何从年薪3w到30w!
查看>>
在.net core上使用Entity FramWork(Db first)
查看>>
System.Net.WebException: 无法显示错误消息,原因是无法找到包含此错误消息的可选资源程序集...
查看>>
UIImage 和 iOS 图片压缩UIImage / UIImageVIew
查看>>
MongoDB的数据库、集合的基本操作
查看>>