Windows下Pytorch的安装及运行

Pytorch的安装分为3个部分:

Anaconda的安装、cuda安装、pytorch安装

Anaconda建议使用最新版本,使用旧版本的,需要升级python等库,会遇到问题。解决问题的时间大于直接重新安装anaconda的时间,所以选择更新anaconda

Anaconda下载地址为:

https://www.anaconda.com/download/

前段时间看到消息,有Python的包已经开始不支持python2了,所以安装的时候最好安装python3,python3在安装时,勾选添加到环境变量的复选框(会将anaconda的python,安装工具conda,pip放到环境变量中,便于后续安装操作)。

Windows下Pytorch的安装及运行

我所使用的anaconda版本及对应的python版本如图所示:

Windows下Pytorch的安装及运行

安装好anaconda,下面安装cuda(cuda的安装,除了硬件的支撑之外,需要一些其他工具的支持,比如vs编译器(vs编译器))

目前的最新cuda版本是9.1,cuda下载网址为:

https://developer.nvidia.com/cuda-downloads

选择操作系统(使用的操作系统为win):

Windows下Pytorch的安装及运行

根据自己的电脑选择(我的是windows、9.1跨平台(Architecture是x86_64)、win版本为8.1):

Windows下Pytorch的安装及运行

我选择本地安装(我比较习惯下载到本地进行安装,避免出现断网,或者自己需要切换网络状态时出现问题,当然在线安装也是可以的)

在安装cuda时,其需要的工具的支持,查看其文档向导:

https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html

Windows下Pytorch的安装及运行

我们需要下面的东西:

1、支持cuda的GPU(买了电脑之后会有,即硬件要求)

2、支持的windows系统(装了windows系统,版本win7以上,也应该都有,xp及以下,装这个的不会还在用xp吧~~~~~~~)

番外:其中windows Server应该是简化版的windows系统,跟有界面的ubuntu和*面的ubuntu之间的区别赶脚差不离(好像腾讯云的windows是windows server,ubuntu是命令行,没有图形界面),之前用windows server的时候感觉与windows系统无异

Windows下Pytorch的安装及运行

3、支持的vs工具,cuda9.1支持的vs工具见下表(上面的向导文档也有),vs2010后面有个depercated,没弄清楚什么鬼~~~~~

Windows下Pytorch的安装及运行

4、Cuda安装包(已有)

Cuda准备就绪,anaconda准备就绪,可以开始pytorch的安装了,pytorch目前官方不支持win,在其主页上是这样的:

Windows下Pytorch的安装及运行

在win下运行,会找不到pytorch和torchvision(如图所示):

Windows下Pytorch的安装及运行

安装pytorch的方法,是在这个知乎专栏里找到的,附上地址,以示对作者的尊重:

https://zhuanlan.zhihu.com/p/26871672

安装的命令有四个,第一个表示没有cuda使用pytorch,也就是只有cpu

其他是安装了对应版本的cuda之后,安装pytorch 的命令(忽略掉注释里的windows版本,只需要看cuda对应的版本就好~,我的cuda版本是9.1.85,最新版本,选择的是第三个命令,安装可用,都是属于cuda9),在安装的时候,最好先墙出去,速度会快些,当然不墙也没关系,就是等的久一点而已嘛~

Windows下Pytorch的安装及运行

这样,pytorch就安装完毕了,但是还有pytorchvision需要安装,在使用时,pytorch和pytorchvision往往需要同时出现,pytorchvision安装方法,我是通过本地安装的方式进行安装的,本地包的下载地址为:https://pypi.python.org/pypi/torchvision/0.1.8

Windows下Pytorch的安装及运行

下载好之后,在当前文件夹下打开命令窗,输入下面的命令,安装pytorchvision:

Windows下Pytorch的安装及运行

至此,所有的安装就已经完成。

下面查看是否安装好pytorch,torchvision和pytorch是否支持cuda,输出为True,证明ok。

Windows下Pytorch的安装及运行