写点什么

PyTorch 深度学习实战 | PyTorch 环境搭建

作者:TiAmo
  • 2023-03-11
    江苏
  • 本文字数:1805 字

    阅读完需:约 6 分钟

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 组合键到控制台,关闭当前图形环境,对应命令如下。

sudo service lightdm stop
复制代码


卸载可能存在的旧版本 NVIDIA 驱动,对应命令如下。

sudo apt-get remove --purge nvidia
复制代码


安装驱动可能需要的依赖,对应命令如下。

sudo apt-get update sudo apt-get install dkms build-essential linux-headers-generic
复制代码


把 nouveau 驱动加入黑名单并禁用 nouveau 内核模块,对应命令如下。

sudo nano /etc/modprobe.d/blacklist-nouveau.conf
复制代码


在文件 blacklist-nouveau.conf 中加入如下内容,对应命令如下。

blacklist nouveauoptions nouveau modeset=0
复制代码


保存后退出,执行,对应命令如下。

sudo update-initramfs -u
复制代码


然后重启,对应命令如下。

reboot
复制代码


重启后再次进入字符终端界面(或按 Ctrl+Alt+F1 组合键),并关闭图形界面,对应命令如下。

sudo service lightdm stop
复制代码


进入之前 NVIDIA 驱动文件下载目录,安装驱动程序,对应命令如下。

sudo chmod u+x NVIDIA-Linux-x86_64-384.98.runsudo ./NVIDIA-Linux-x86_64-384.98.run -no-opengl-files
复制代码


-no-opengl-files 表示只安装驱动文件,不安装 OpenGL 文件。这个参数不可忽略,否则会导致登录界面死循环。

最后重新启动图形环境,对应命令如下。

sudo service lightdm start
复制代码


通过以下命令确认驱动是否正确安装,对应命令如下。

cat /proc/driver/nvidia/version
复制代码


至此,NVIDIA 显卡驱动程序安装成功。

2●PyTorch 安装

进入 PyTorch 官网https://pytorch.org

如图 2 所示,根据 CUDA 和 Python 的版本以及平台系统等找到适合 PyTorch 的版本,之后会自动提示“Run this command” 命令指令,将指令复制到命令行,进行安装。

■ 图 2 PyTorch 官网


3●安装 torchvision

安装好 PyTorch 后,还需要安装 torchvision。torchvision 中主要集成了一些数据集、深度学习模型、一些转换等,在使用 PyTorch 的过程中是不可缺少的部分。

安装 torchvision 比较简单,可直接使用 pip 命令安装。

pip install torchvision
复制代码


4●更新 NumPy

安装成功 PyTorch 和 torchvision 后,打开 iPython,输入: 

import torch
复制代码


此时可能会出现报错的情况,报错信息如下。

ImportError: numpy.core.multiarray failed to import
复制代码


这是因为 NumPy 的版本需要更新,直接使用 pip 命令更新 NumPy,对应命令如下。

pip install 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 部分一致,请翻阅查看操作!


发布于: 刚刚阅读数: 3
用户头像

TiAmo

关注

有能力爱自己,有余力爱别人! 2022-06-16 加入

CSDN全栈领域优质创作者,万粉博主;阿里云专家博主、星级博主、技术博主、阿里云问答官,阿里云MVP;华为云享专家;华为Iot专家;亚马逊人工智能自动驾驶(大众组)吉尼斯世界纪录获得者

评论

发布
暂无评论
PyTorch深度学习实战 | PyTorch环境搭建_PyTorch_TiAmo_InfoQ写作社区