如何安装gpu版本的pytorch（超详细）

目录

1.安装CUDA

2.安装pytorch

1、检查自己电脑的gpu与cpu

2.查询你的python版本

3.安装pytorch

1、进入官方页面pytorch

2、pytorch的whl下载地址

3.检验pytorch安装是否完成

---

1.安装CUDA

首先，在命令行输入nvcc -V检验是否已安装CUDA

若结果显示为

则表明已经配置过CUDA。

若结果为

则尚未配置，配置过程可见笔者的另一篇文章CUDA安装教程（超详细Windows版）

2.安装pytorch

1、检查自己电脑的gpu与cpu

在左下角搜索栏中输入设备管理器，从中找到显示适配器即可看到电脑里的gpu与cpu了。

这里就可以看出你是否有gpu了。

2.查询你的python版本

ctrl+r后输入python

3.安装pytorch

1、进入官方页面pytorch

找到这个页面

从中根据我们的 版本在Run this Command中找到其所对应的网址

https://download.pytorch.org/whl/cu117从中找到我们所要安装的torch torchvision torchaudio

会相对复杂一点。

2、pytorch的whl下载地址

直接进入这个页面开始查找

这里cpu是安装cpu版本的cu是gpu版本的，cp指的是你的python版本，还有就是win和linux系统的选择，找到你所对应的版本将其点击下载。

找到这三个文件将其地址复制到你python的配置环境位置。

在这里进入命令窗口或者cd到这里

对这三个文件分别pip安装，

要按照torch，torchvision，torchaudio的顺序安装完成。

3.检验pytorch安装是否完成

如果结果为True则安装成功。

​​​​​​​​​​​