PyTorch 深度学习实战 | PyTorch 环境搭建
PyTorch 是当前主流深度学习框架之一,其设计追求最少的封装、最直观的设计,其简洁优美的特性使得 PyTorch 代码更易理解,对新手非常友好。
本文主要介绍 PyTorch 环境搭建!
1、Linux 平台下 PyTorch 环境搭建
下面以 Ubuntu 16.04 为例,简要讲述 PyTorch 在 Linux 系统下的安装过程。在 Linux 平台下,PyTorch 的安装总共需要 5 个步骤,所有步骤内的详细命令皆已列出,读者按照顺序输入命令即可完成安装。
1●安装显卡驱动
如果需要安装 CUDA 版本的 PyTorch,计算机也有独立显卡,则需要更新 Ubuntu 独立显卡驱动。否则即使安装了 CUDA 版本的 PyTorch 也无法使用 GPU。
如图 1 所示,进入官网https://www.nvidia.com/Download/index.aspx?lang=en-us,查看适合本机显卡的驱动,下载 runfile 文件,如 NVIDIA-Linux-x86_64-384.98.run。
■ 图 1 NVIDIA 官网
下载完成后,按 Ctrl+Alt+F1 组合键到控制台,关闭当前图形环境,对应命令如下。
卸载可能存在的旧版本 NVIDIA 驱动,对应命令如下。
安装驱动可能需要的依赖,对应命令如下。
把 nouveau 驱动加入黑名单并禁用 nouveau 内核模块,对应命令如下。
在文件 blacklist-nouveau.conf 中加入如下内容,对应命令如下。
保存后退出,执行,对应命令如下。
然后重启,对应命令如下。
重启后再次进入字符终端界面(或按 Ctrl+Alt+F1 组合键),并关闭图形界面,对应命令如下。
进入之前 NVIDIA 驱动文件下载目录,安装驱动程序,对应命令如下。
-no-opengl-files 表示只安装驱动文件,不安装 OpenGL 文件。这个参数不可忽略,否则会导致登录界面死循环。
最后重新启动图形环境,对应命令如下。
通过以下命令确认驱动是否正确安装,对应命令如下。
至此,NVIDIA 显卡驱动程序安装成功。
2●PyTorch 安装
进入 PyTorch 官网https://pytorch.org,
如图 2 所示,根据 CUDA 和 Python 的版本以及平台系统等找到适合 PyTorch 的版本,之后会自动提示“Run this command” 命令指令,将指令复制到命令行,进行安装。
■ 图 2 PyTorch 官网
3●安装 torchvision
安装好 PyTorch 后,还需要安装 torchvision。torchvision 中主要集成了一些数据集、深度学习模型、一些转换等,在使用 PyTorch 的过程中是不可缺少的部分。
安装 torchvision 比较简单,可直接使用 pip 命令安装。
4●更新 NumPy
安装成功 PyTorch 和 torchvision 后,打开 iPython,输入:
此时可能会出现报错的情况,报错信息如下。
这是因为 NumPy 的版本需要更新,直接使用 pip 命令更新 NumPy,对应命令如下。
至此,PyTorch 安装成功。
5●测试
输入如图 3 所示的命令后,若无报错信息,说明 PyTorch 已经安装成功。输入如图 4 所示的命令后,若返回为“True”,说明已经可以使用 GPU。
■ 图 3 测试命令行截图 1
■ 图 4 测试命令行截图 2
2、Windows 平台下 PyTorch 环境搭建
从 2018 年 4 月起,PyTorch 官方开始发布 Windows 版本。在此简要讲解在 Windows 10 系统下,安装 PyTorch 的步骤。鉴于已经在前文中讲述了显卡驱动程序在 Linux 系统下的配置过程,Windows 系统下的配置也基本相似,所以不再单独讲述显卡驱动在 Windows 系统下的配置过程。
PyTorch 在 Windows 系统上的安装主要有两种方法:通过官网安装,conda 安装(本机上需要预先安装 Anaconda|Python)。
1●通过官网安装
进入官网https://PyTorch.org/,如图 5 所示。
■ 图 5 PyTorch 官网截屏图
如前文介绍的 Linux 系统下安装一样,根据 CUDA 和 Python 的版本以及平台系统等找到适合 PyTorch 的版本,之后会自动提示“Run this command” 命令指令,将指令复制到命令行,进行安装。
2●conda 安装 PyTorch 包
在 Windows 的命令行输入图 6 中框内的命令(请注意控制 CUDA 版本和 CPU/GPU 版本),等待一段时间后,出现图 6 中的输出后,即完成了安装。
■ 图 6 conda 安装命令行截屏图
安装完成后,同样需要安装 torchvision,具体方法与上述 Linux 部分一致,请翻阅查看操作!
版权声明: 本文为 InfoQ 作者【TiAmo】的原创文章。
原文链接:【http://xie.infoq.cn/article/8e5eb5f1b6a09bfbd53abce10】。文章转载请联系作者。
评论