深度学习实践教程
上QQ阅读APP看本书,新人免费读10天
设备和账号都新为新人

2.2 Windows系统中PyTorch的配置

2.2.1 安装Python

PyTorch是在Python的基础上安装的,所以首先要在计算机上安装Python。本书安装的Python版本为Python 3.5.3。以下为在Window系统中安装Python的简单步骤。

打开浏览器访问Python官方网站,进入下载界面,网址为https://www.python.org/downloads/windows/,如图2.1所示。

图2.1 Python 3.5.3下载界面

双击下载的安装文件,进入Python安装向导,如图2.2所示。

图2.2 Python安装向导

勾选Add Python 3.5 to PATH,将Python添加到环境变量中。单击“Install Now”按钮,按照安装向导的提示进行安装。

安装成功界面如图2.3所示。

图2.3 Python安装成功界面

2.2.2 PyTorch环境搭建

登录PyTorch官方网站https://pytorch.org/,如图2.4所示。

图2.4 PyTorch官方网站

单击“Get Started”按钮后进入PyTorch安装界面,如图2.5所示。这里介绍pip安装方式。图2.5中右侧区域为计算机系统及CUDA版本的选择,计算机的配置决定了“Run this Command”后面对应的安装命令,复制“Run this Command”后面对应的安装命令。

图2.5 PyTorch安装界面

打开cmd命令行窗口,粘贴刚刚复制的安装命令,然后按回车键。

若出现图2.6所示的结果,则表示安装成功。

图2.6 PyTorch安装成功

下一步为安装torchvision。torchvision是一个很重要的、功能强大的视觉工具包。torchvision由流行的数据集(torchvision.datasets)、模型架构(torchvision.models)和用于计算机视觉的常用图像转换函数(torchvision.transforms)组成。

安装torchvision的方法是在cmd命令行窗口中输入如下命令:

然后检验Python是否安装成功,在cmd命令行窗口中输入:

如果安装成功,则会出现提示符“>>>”,表示进入了Python环境。

执行代码:

若结果如图2.7所示,则说明Windows系统中的PyTorch环境搭建成功。

图2.7 PyTorch环境搭建成功