目录
1 准备阶段
1.1 安装Anaconda
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
chmod +x Miniconda3-latest-Linux-x86_64.sh
./Miniconda3-latest-Linux-x86_64.sh方法一：官网下载Anaconda安装。
方法二： 官网下载太慢，推荐国内镜像源下载最新版：
安装过程可以参考：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 下载镜像文件
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 检验是否安装成功
同方案一验证