Win10系统pytorch安装教程

前提：安装后Anaconda和NVIDIA显卡驱动

1. 确定支持的最高CUDA版本

确定支持的最高CUDA版本
记住这个CUDA版本，后面要用。

2. 新建conda虚拟环境

2.1 为conda配置清华镜像源

conda常用命令
打开Anaconda prompt窗口，使用如下命令：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

2.2 新建指定python版本的conda虚拟环境

新建python 3.9版本的conda虚拟环的境命令：

conda create -n pytorch python=3.9

切换到新建的conda虚拟环境

conda activate pytorch

3. 在conda虚拟环境安装pytorch

3.1 确定要安装的pytorch版本

python和pytorch版本对应关系

举个例子，我选的python版本3.9，最高支持cuda版本为12.2：

torch 1.11及以上都是可以的
这里我选择torch 1.12.1版本

pytorch版本和cuda版本的对应关系

在这个页面找到对应torch 1.12.1版本，根据cuda版本（不能超过第1节中的cuda最高版本）复制对应的命令，在Anaconda prompt中执行：

执行结果如图，则表示成功；如果下载速度很慢，可以更换国内镜像源后，重新执行。

使用如下命令查看刚才新建的conda虚拟环境已安装的包

conda list

4. 测试是否安装成功

依次执行如下命令：

python
import torch
torch.cuda.is_available()

如果输出为True表示安装pytorch成功