PyTorch安装教程（GPU版）

目录

1 准备阶段

1.1 安装Anaconda

1.2 查看显卡型号

1.3 更新最新版本显卡驱动

1.4 查看显卡驱动版本信息

2 安装PyTorch（GPU版）

2.1 通过conda创建一个虚拟环境

2.2 方案一：通过官网命令行安装Pytorch

2.2.1 配置官网命令（推荐pip）

2.2.2 在创建的虚拟环境中通过命令行自动下载安装

2.2.3 检查是否安装成功

2.3 方案二：通过本地whl文件的方式安装

2.3.1 下载镜像文件

2.3.2 安装

2.3.3 检验是否安装成功

---

1 准备阶段

1.1 安装Anaconda

方法一：官网下载Anaconda安装。

方法二： 官网下载太慢，推荐国内镜像源下载最新版：

• anaconda清华大学开源软件镜像站

• Anaconda3

安装过程可以参考：Anaconda安装-超详细版(2023)

1.2 查看显卡型号

右击开始菜单→任务管理器→性能→GPU 1，其中 NVIDI GeForce GTX 1050表示GPU型号。 

注：若只显示：GPU 0   Intel(R) HD Graphics 630，是CPU集成显卡，表示没有GPU，不能安装GPU版本的PyTorch。

1.3 更新最新版本显卡驱动

在官网下载对应型号的最新显卡驱动，并安装(全部下一步)，地址：NVIDIA GeForce 驱动程序

安装结束后，重启电脑。

1.4 查看显卡驱动版本信息

双击NVIDIA设置小图标打开NVIDIA控制面板（或桌面空白处鼠标右击→NVIDA控制面板）→帮助→系统信息→组件，查看电脑驱动的版本：NVIDIA CUDA 12.6.32（12.6版）。

注：后期安装PyTorch时，选取的CUDA不能高于12.6。

2 安装PyTorch（GPU版）

2.1 通过conda创建一个虚拟环境

按住win＋R ，输入cmd，在命令行输入：conda create –n 虚拟环境名字 python=版本，即可创建成功。本部分创建的虚拟环境名称为：myPytorch，python版本为3.8。

conda create –n myPytorch python=3.8

查看已创建的虚拟环境：

conda env list

2.2 方案一：通过官网命令行安装Pytorch

2.2.1 配置官网命令（推荐pip）

官网地址： PyTorch  或者 Start Locally | PyTorch

注：此处CUDA的版本 ≤ 电脑显卡驱动版本（最好保持一致），可以通过更新电脑显卡驱动（前面提过）或者在Previous versions of PyTorch 中寻找低版本的CUDA。

2.2.2 在创建的虚拟环境中通过命令行自动下载安装

打开Anaconda Prompt命令行窗口：开始→Anaconda Prompt （或者win+R，输入cmd）。

进入虚拟环境：输入conda activate 虚拟环境

conda activate myPytorch

在虚拟环境下安装，输入官网配置的命令：

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

2.2.3 检查是否安装成功

方案一：输入pip list或者conda list，看有没有pytorch或者torch，有表示成功。

方案二：通过python验证

在命令行以此输入：
python
import torch
torch.cuda.is_available()，这个命令是检查我们pytorch的GPU能否用。

python
import torch
torch.cuda.is_available()

如果显示True，就说明PyTorch安装成功了。

2.3 方案二：通过本地镜像文件离线方式安装（推荐清华镜像源）

2.3.1 下载镜像文件

• https://download.pytorch.org/whl/torch/
  

• 清华大学开源软件镜像站
  
  

2.3.2 安装

打开Anaconda Prompt命令行窗口：开始→Anaconda Prompt （或者win+R，输入cmd）。

切换到虚拟环境：conda activate 虚拟环境     

方案一：找到下载好的.whl文件，在命令行运行：

pip3 install C:\Users\Lenovo\Desktop\torch-2.4.0+cu124-cp38-cp38-win_amd64.whl

注：换成 pip3 install 你的路径

方案二：清华镜像源，下载pytorch和torchvision，在命令行运行：

conda install --offline C:\Users\Lenovo\Desktop\pytorch-2.4.0-py3.8_cuda12.4_cudnn9_0.tar.bz2
conda install --offline C:\Users\Lenovo\Desktop\torchvision-0.19.0-py38_cu124.tar.bz2

注：换成
                conda install --offline 路径+pytorch压缩包的全称
                conda install --offline 路径+torchvision压缩包的全称

2.3.3 检验是否安装成功

同方案一验证