1.一个简单的介绍

本文档介绍pytorch在Windows10上的安装过程，pytorch的官方网站为https://pytorch.org/。建议以后所有学习和安装操作都以官网为准。pytorch的安装可以有以下类型：
官方推荐使用conda来安装pytorch，但是本文使用的是python的pip包管理器，libtorch仅支持C++，source是源代码安装，对于初学者并不友好。我们选择语言为Python。Pytorch有CPU版本和CUDA版本，作者电脑安装的是CUDA11.1。选择完成后，自动出现安装命令如下：

pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

2. 下载包

执行上面的命令可能耗时较久，可以使用IDM下载下面的两个包在下面的链接上： https://download.pytorch.org/whl/torch_stable.html

严格下载官方提供的版本号，否则会出现问题。

3. 更新NVDIA显卡驱动

因为CUDA是NVIDIA的技术，所以只支持NVIDIA显卡。更新显卡驱动是为了匹配更高版本的CUDA，从而匹配更高版本的pytorch。需要知道自己显卡型号，笔者的型号为Geforce RTX 2060(notebook)。下载驱动从： https://www.nvidia.cn/Download/index.aspx?lang=cn

4. 安装CUDA和CUDNN

参考https://blog.csdn.net/qq_23013309/article/details/103965619

5. 安装pytorch

在下载目录中打开cmd，执行以下命令即可

pip install torch-1.9.0+cu111-cp39-cp39-win_amd64.whl
pip install torchvision-0.10.0+cu111-cp39-cp39-win_amd64.whl

6. 测试

新建test.py，输入以下代码：

import torch
print('torch version {}'.format(torch.__version__))
print('cuda is available {}'.format(torch.cuda.is_available()))

执行结果如下，表明安装成功。

Linux