Linux下怎么安装PyTorch

共计 2406 个字符，预计需要花费 7 分钟才能阅读完成。

本篇内容主要讲解“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 网站，更多相关内容可以进入相关频道进行查询，关注我们，继续学习！

向 AI 问一下细节

丸趣 TV 网 – 提供最优质的资源集合！