【PyTorch学习】安装PyTorch

安装

PyTorch 安装起来很简单，官网就有不同版本的安装教程：

只需要根据实际情况选择适合的安装方法就好了。

例如，使用的是 Windows 系统，想用 pip 安装，Python 是 3.6 版的，没有 GPU 加速，那就按上面的选，然后根据上面的提示，在 Terminal 中输入以下指令就好了：

pip3 install torch==1.3.1+cpu torchvision==0.4.2+cpu -f https://download.pytorch.org/whl/torch_stable.html

torch

torch 是主模块, 用来搭建神经网络的。

torchvision

torchvision 是辅模块, 有数据库, 还有一些已经训练好的神经网络等着你直接用, 比如 (VGG, AlexNet, ResNet)。