操作系统:win10
已安装程序:Python 3.6 + Anaconda 5.1.0 + CUDA 9
pytorch官网:https://pytorch.org/
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.whl pip3 install torchvision
//pip3不能用换成pip
3、Source
根据官网提示安装,网址:https://github.com/pytorch/pytorch#from-source,个人认为没有前两种方便,毕竟cmd下可直接安装。
第一次我使用conda,在cmd命令行下安装,输入conda install pytorch torchvision -c pytorch,报错:找不到对应的包,就转为使用pip方法,后来查了一下,可以手动找包,参考Blog:https://blog.csdn.net/muyimo/article/details/84567630 和 https://blog.csdn.net/ksws0292756/article/details/79192268。(此种方法未经尝试,仅供参考)
接下来说一说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的方法 http://www.mamicode.com/info-detail-2424578.html,执行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里正常使用。