【pytorch笔记】（一）安装

深度学习框架有很多，如谷歌的TensorFlow，百度的PaddlePaddle，亚马逊的MXNet等。在众多框架中，我最喜欢的还是Facebook推出的pytorch，与TensorFlow相比，pytorch更加简洁，易调试，容易上手。

（图源：水印）

之后会开始尝试更新我在pytorch学习之路上的一些学习笔记，欢迎大家监督。时间有限，今天先写篇在win10环境下的pytorch简略安装步骤。

首先，安装前必须要知道pytorch官网。

之后安装命令可以直接在官网“定制”。记不住官网网址也没关系，直接百度搜pytorch即可。

CPU环境

CPU环境安装起来最为简单，适合新手入门、不想折腾、电脑没英伟达显卡的用户。

先点击上文pytorch官网的install按钮，跳转到以下界面：

选择好自己电脑所对应的系统和环境，然后最下面一行即为安装命令，

pip3 install torch torchvision torchaudio

将它复制粘贴，win + Q 搜索打开anaconda prompt，粘贴一下命令，敲一下回车即可安装。

GPU环境

如果你有一块显存较大的英伟达显卡（打开任务管理器→性能→GPU栏里有NVIDIA Geforce字样），那么恭喜你可以获得显卡加速，提升计算速度，让深度学习像呼吸一样自然。

GPU环境下安装步骤比较多，而且需要安装的依赖软件比较多，以下给出安装流程：

1. Visual Studio;
2. NVIDIA CUDA;
3. NVIDIA cudnn;
4. pytorch-gpu.

前3项的具体安装步骤可以参考这篇博客。

前3项都完成后，我们就可以安装GPU版本的pytorch了。依旧回到pytorch官网界面：

选择好安装的cuda对应版本的pytorch，并执行官网给的代码。

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

如果下载速度太慢，可以尝试更换镜像。

conda更换镜像教程。

测试

接下来做个简单的小测试来确认pytorch是否安装成功。

打开anaconda prompt，逐行输入以下代码：

python 
import torch
x = torch.ones(5)
x

出现以上结果那么恭喜你安装成功pytorch！