共计 2381 个字符，预计需要花费 6 分钟才能阅读完成。
本篇内容主要讲解“Linux 下怎么安装 PyTorch”，感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷，实用性强。下面就让丸趣 TV 小编来带大家学习“Linux 下怎么安装 PyTorch”吧!
一、PyTorch 简介
PyTorch 是一个开源的 Python 机器学习库，基于 Torch，用于自然语言处理等应用程序。2017 年 1 月，由 Facebook 人工智能研究院（FAIR）基于 Torch 推出 PyTorch。PyTorch 的前身是 Torch，其底层和 Torch 框架一样，但是使用 Python 重新写了很多内容，不仅更加灵活，支持动态图，而且提供了 Python 接口。它是由 Torch7 团队开发，是一个以 Python 优先的深度学习框架，不仅能够实现强大的 GPU 加速，同时还支持动态神经网络。PyTorch 既可以看作加入了 GPU 支持的 numpy，同时也可以看成一个拥有自动求导功能的强大的深度神经网络。除了 Facebook 外，它已经被 Twitter、CMU 和 Salesforce 等机构采用。
二、安装步骤 1、操作系统选择
emsp; emsp; 查看 PyTorch 官网，可以看到 PyTorch 支持 Linux、Mac、window 平台、支持 conda、pip、源码等安装方式，也支持 CPU、cuda、ROCm 计算平台，我们点击环境选择可以发现目前只有 linux 系统是支持全语言、全安装方式、全计算平台的，所以我们选择 linux 操作系统作为系统环境。另外机器学习计算要求 glibc 版本要求较高，centos 搭载的内核和 glbic 版本较低，Ubuntu 搭载的内核版本都较新，所以机器学习主机建议使用 Ubuntu 操作系统。目前 cuda 更新支持的最低 Ubuntu 版本为 18.04，所以建议使用 Ubuntu18.04 以上的操作系统。
wuhs@s169:~$ cat /etc/os-release
NAME=“Ubuntu”
VERSION=“18.04.6 LTS (Bionic Beaver)”
2、Anaconda3 安装
emsp; emsp; 如上所示，PyTorch 支持的安装方式有多种，博主拟采用 conda 安装方式，建议先安装 Anaconda3，可以根据我们需要创建不同虚拟环境，虚拟环境下安装不同的机 PyTorch 版本，虚拟环境支持互不影响。Ubuntu 环境下 anaconda 的安装见博文 Ubuntu 之 Anaconda3 安装。
wuhs@s169:~$ wget https://mirrors.bfsu.edu.cn/anaconda/archive/Anaconda3-2022.10-Linux-x86_64.sh
wuhs@s169:~$ sh Anaconda3-2022.10-Linux-x86_64.sh
wuhs@s169:~$ source ~/.bashrc
3、查看 Python 版本
emsp; emsp; 不同的 PyTorch 版本要求的 Python 版本是不一样的，所以安装好 anaconda3 后我们检查当前的 Python 版本，默认初始化都是当前 anaconda3 发布时对应的 Python 最新版本，当然我们也可以使用 conda 创建所需的 Python 环境版本。我们在 torchvision 查看 PyTorch、torchvision、Python 版本匹配要求。
(base) wuhs@s169:~$ python -V
Python 3.9.13
4、安装 PyTorch
emsp; emsp; 如下第二步，PyTorch 官网我们可以在选择操作系统、安装方式、编程语言、计算平台后生成对应的安装命令。
(base) wuhs@s169:~$ conda install pytorch torchvision torchaudio cpuonly -c pytorch
Collecting package metadata (current_repodata.json): done
Solving environment: done
## Package Plan ##
Proceed ([y]/n)? y
hellip;
5、版本验证
(base) wuhs@s169:~$ python
Python 3.9.13 (main, Aug 25 2022, 23:26:10)
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type“help”,“copyright”,“credits”or“license”for more information.
import torch
torch.version
lsquo;1.13.1 rsquo;
三、指定版本安装 1、创建虚拟环境
(base) wuhs@s169:~$ conda create -n pytorch python=3.9
hellip;
(base) wuhs@s169:~$ conda activate pytorch
(pytorch) wuhs@s169:~$
2、安装指定版本的 PyTorch
emsp; emsp; 安装指定版本的 PyTorch 的时候我们需要在 GitHub 官网 PyTorch 频道查看匹配版本，conda 安装的时候指定版本号，具体版本号可以查看 anaconda 官网，PyTorch 版本和 TorchAudio 对应关系见
TorchAudio。当然如果我们指定版本错误的情况下，安装的时候会报错，我们根据报错提示核验是哪个软件版本指定错误，再去官网合适确认修正后重新安装即可。
(pytorch) wuhs@s169:~$ conda install pytorch3.12.0 torchvision=0.13.0 torchaudio0.12.0 cpuonly -c pytorch
到此，相信大家对“Linux 下怎么安装 PyTorch”有了更深的了解，不妨来实际操作一番吧！这里是丸趣 TV 网站，更多相关内容可以进入相关频道进行查询，关注我们，继续学习！
