PyTorch安装与配置教程（2022.11）

1. PyTorch的安装

（1）首先在命令行输入 nvidia-smi 查看本机的 CUDA 版本：

（2）前往 PyTorch 官网：PyTorch，在 Get Started 中设置以下选项：

如果想在自己电脑上跑通代码，就选 CUDA，如果不需要在自己电脑上跑，而是在服务器上跑，或者没有独立显卡，就选 CPU。独立显卡需要 NVIDIA 显卡。这里我们一定要选择和自己版本相同或更低的 CUDA。

（3）激活一个 Anaconda 环境（本文在 PyTorch 环境下操作），Anaconda 的安装与使用可以转至：Anaconda3安装与配置教程（2022.11），由于直接用 PyTorch 官网给出的命令进行安装速度非常慢，还很容易出错，可以先修改镜像源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/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 --show

如果需要将镜像源恢复成默认设置的话可以用以下命令：

conda config --remove-key channels

（4）安装 PyTorch（注意需要将 -c pytorch 去掉，不然还是强制使用官网进行下载的）：

conda install pytorch torchvision torchaudio pytorch-cuda=11.6

（5）如果还是下载不了的话可以直接前往清华大学镜像源：清华大学 PyTorch 镜像源，手动下载 pytorch、torchvision 以及 torchaudio，本文最后使用的就是这种方法：

注意：py 和 cu 后面的数字分别表示 Python 和 CUDA 的版本号，找到对应版本进行下载即可。

下载好后进行离线安装：

conda install --offline D:/pytorch-1.13.0-py3.9_cuda11.6_cudnn8_0.tar.bz2conda install --offline D:/torchvision-0.14.0-py39_cu116.tar.bz2conda install --offline D:/torchaudio-0.13.0-py39_cu116.tar.bz2

（6）前往 CUDA Toolkit Archive 下载对应版本的 CUDA 套件（注意如果电脑是 Win10，Version 需要选10）：

下载好后打开程序安装 NVIDIA GPU Computing Toolkit，安装时路径需要使用默认的，即 C:\Program Files\NVIDIA GPU Computing Toolkit 以及 C:\Program Files\NVIDIA Corporation。

安装好后打开命令行，检查版本：

nvcc -V

（7）进入 Anaconda 环境，安装剩余的包：

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

（8）进入 Anaconda 环境，打开 Python，输入以下内容进行测试，没有报错即安装成功：

>>> import torch>>> torch.cuda<module 'torch.cuda' from 'D:\\Anaconda3_Environments\\envs\\PyTorch\\lib\\site-packages\\torch\\cuda\\__init__.py'>>>> torch.cuda.is_available()True>>> from __future__ import print_function>>> x = torch.rand(5,3)>>> print(x)tensor([[0.4175, 0.7341, 0.7712],[0.3714, 0.4031, 0.8727],[0.3453, 0.1515, 0.6602],[0.6127, 0.2680, 0.4209],[0.0118, 0.3593, 0.7251]])