PyTorch安装教程

1、安装anaconda

安装过程与fbprophet安装步骤相同，见Anaconda安装教程

2、显卡配置

如果不带显卡，可以略过这一步

打开任务管理器，GPU那里看到了你的 NVIDIA 显卡，即可

 3、在anaconda里创建安装环境

打开Anaconda prompt，输入以下命令，env_pytorch是环境的名字，3.6是python版本

Proceed ([y]/n)? y  出现这个 输入y即可

conda create -n env_pytorch python=3.6

可以使用命令conda info --envs看下你的环境

激活环境

conda activate env_pytorch

之后就进入了我们创建的环境env_pytorch中，即是前面括号里的base变成了env_pytorch

4、安装PyTorch

 打开PyTorch官网，下拉找到下面的界面

PyTorch Build 选择 Stable、系统根据自己选择、Package，Windows下推荐 conda，Linux 下推荐 pip；Language我选择Python，CUDA 推荐10.2。我的选择如图所示

 复制下面那一段代码，到Anaconda prompt命令中

（一定要在刚刚我们创建的环境env_pytorch中执行）

接下里就是漫长的等待，其实还好我就几分钟就下载好了

5、验证是否安装成功

在env_pytorch输入python，进入python

之后输入import torch，如果没有报错说明PyTorch 已经安装上了

接下来，输入torch.cuda.is_available()，如果是 True，说明可以使用 GPU，如果是 False，说明只能使用CPU。