Pytorch安装

一、总体的步骤

1
2
3
4
5
6
7
8
9
10
11
12
13
Windows
  ↓
[可选步骤] 安装 Anaconda（或Miniconda，推荐）
  ↓
安装 Python 环境
  ↓
去 Pytorch 官网找到对应的下载链接
  ↓
选择 CPU版 / CUDA版
  ↓
pip 安装 PyTorch
  ↓
验证是否安装成功

二、安装 conda 环境

详细步骤：https://999314.xyz/posts/ad482a7d/

三、conda 环境里安装 python 环境

1
2
3
conda create -n dl python=3.13 -y  # 注册一个新环境，并安装 3.13.x 版本的python环境
conda activate dl  # 激活环境
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu126  # 下载 pytorch

若不愿意按装 conda 环境，请跳过看下一步

四、去 Pytorch 官网找下载链接

先确认自己电脑的信息：

1
nvidia-smi

显示的 CUDA UMD Version 是你最高可安装的 cuda 版本
05cbede4-8b46-4db1-8d5f-5d5f608b455e.png

去到官网：https://pytorch.org/，这里就需要用到魔法了，因为裸连几乎进不去。

五、pip 安装 torch 包

步骤请看图👇
251af942-ba29-44dc-87e7-fec6a53e3d2c.png
f37d91b0-d7bb-4deb-86bf-f8a64b90c165.png

选择 CUDA 版本时不能大于上面第一张图输出的 CUDA UMD Version 版本。

看图👇
33326236-62a7-4c3e-9f04-94e49311adf1.png

安装成功是这样的，无爆红：
17329e8c-3614-4be4-9ada-f604ef22d4d5.png

Tip

如果命令行里安装不下来，出现爆红的现象，这是正常的，开启小猫咪的虚拟网卡模式（也叫 TUN 模式）即可

六、验证

1
2
3
4
5
6
import torch

print(torch.__version__)
print(torch.rand(3, 3))

print(f"验证GPU版是否可用：{torch.cuda.is_available()}")

在编辑器中选择安装了 torch 的环境，可以点击右下角的 选择解释器，或者按住 Ctrl + Shift + P 输入：Python:选择解释器 选择即可

daed9dce-6fbe-4701-8c82-f3fc825e595a.png
16c03316-b13b-472b-a33e-f05648799bad.png

至此安装过程就结束。选择 conda 管理环境的意义就是哪天不想用某个环境了，直接删除就可以了，不影响其他已安装的环境

删除环境的命令如下：

1
conda remove -n <环境名> --all