【python环境之torch】为现有环境安装pytorch、torchvision和torchaudio

Where1s

于 2024-04-11 18:57:10 发布

阅读量2.8k

点赞数 10

文章标签： python pytorch 开发语言

本文链接：https://blog.csdn.net/salaryresist/article/details/137582327

版权

【深度学习环境之torch】为现有环境安装torch、torchvision、torchaudio，以cu117+python3.9为例

由于大多数帖子在编写安装教程时都是从安装显卡驱动和cuda开始。但是在现实应用中，大多数情况下电脑都已经安装完cuda和显卡驱动且代码需要在固定python版本下运行。
针对以上问题，本篇以cu117+python3.9为例介绍如何为现有环境安装pytorch等python库。本篇共介绍三种安装方式，分别是，pip安装、离线安装和conda安装。

总述（重要！！重要！！重要！！！）

由于torch与操作系统、python版本和cuda版本有关 ，所以无论是pip安装、离线安装还是conda安装，安装的流程均为

查看cuda版本->根据对应版本查找pytorch对应文件->下载安装包并安装

1、pip安装

作为常用的安装pytorch 的方法，使用pip安装主要包含以下几步：

1、查看电脑上的cuda版本（以linux系统为例）

使用如下指令查看自己电脑上的cuda版本：

nvcc-V

输入后出现以下界面，图中红框中的内容为电脑的cuda版本。
在这里插入图片描述

注意！！！！！！！在实际应用中，常使用如下指令查看电脑显卡的运行情况

nvidia-smi

在这里插入图片描述

图中标红处也显示cuda版本。但是，此cuda版本并不是系统安装的cuda版本，而是当前安装的显卡驱动可安装的最高cuda版本。 因此，尽量不要盲目使用nvidia-smi指令查看电脑安装的cuda版本。

2、查找需要安装的pytorch版本。

（1）进入pytorch官网：

https://pytorch.org/

在这里插入图片描述
点击Previous versions of Pytorch ，在出现的界面中根据第一步的cuda版本选择对应需要的pytorch版本。此处以cuda11.7为例，cuda117可安装的pytorch最低版本为v1.13.0，最高版本为v2.0.1。根据官方介绍pytorch2可兼容pytorch1系列的所有内容，因此此处建议安装v2.0.1。

在这里插入图片描述
找到对应操作系统，对应cuda和对应pytorch版本的pip安装指令，此处以Linux操作系统，cuda11.7，pytorch2.0.1为例，可以查到对应的pip安装指令为

pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2

3、安装pytorch

打开命令行，使用如下指令进入需要安装pytorch的环境中：

conda activate xxx ##xx代表需要安装的具体环境名称

进入对应环境后，输入下面的指令安装torch，torchvision和torchaudio。

pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2

输入指令后，系统会自动检查相同python版本的其他环境中是否具有需要安装的torch文件。如果存在，将自动复制所有文件至目标文件夹，完成安装。
若没有搜索到与之对应的文件，将自动下载所有文件并进行安装。由于下载地址为外网地址，所以下载的过程相对较慢，耐心等待即可。若嫌下载速度过于缓慢，可自行加速加载。

注意！！这里不建议采用换源的方法，使用国内源下载。国内源并没有包含所有pytorch版本，可能无法下载。即使有对应版本可供下载，也可能导致安装后的文件出现不知名bug，影响使用。如果没有任何办法，可采用下面的离线安装方法安装pytorch。

4、验证是否安装成功

通过在命令行中输入以下指令验证pytorch是否安装成功

python
>>>import torch
>>>torch.cuda.is_available()
True

当显示True表示torch安装成功，其他包测试方法同上。

2、离线安装

当使用pip方法过慢或无法下载安装包时，可使用离线安装方法进行安装。安装的步骤与pip类似，也分为，查看cuda版本、查找并下载安装包和安装三步。

1、查看cuda版本

此操作步骤与pip安装方法相同，可查看上面关于cuda的查看方法。

2、下载对应安装包

当查看完对应的cuda版本后，点击如下网址进入网络查找并下载安装包。此处提供两个下载地址一共选择，选择其一即可

https://download.pytorch.org/whl/torch/
https://download.pytorch.org/whl/torch_stable.html

进入界面后可以发现，网页中提供了大量的torch版本以供用户下载。想要在其中找到自己需要的版本是一件非常困难的事情。此处对如何查找想要的whl文件进行说明。
** 例如，我们需要下载Linux操作系统，cuda=11.7，python=3.9的torch文件** ，所以需要找到下面这个文件。
在这里插入图片描述
这里介绍一下对应规则：

注意！！！！
torch-2.0.0：代表torch版本，还有torchvision和torchaudio
cu117：代表cuda版本号，通过第一步查询得到
cp39：代表python版本。此处python版本为3.9，所以选择“cp39”
linux_x86_64：代表操作系统，具有linux_x86_64，win_amd64，cpu和macos四种

之后下载相应的whl文件。一共要下载三个文件，分别是torch，torchvision和torchaudio。torchvision和torchaudio可能与torch并不在一起，向下多翻几次便可找到。

3、安装