Linux anaconda 环境下安装pytorch

安装之间可以先看一下以下教程，方便安装
Linux系统安装Anaconda3保姆级教程
Anaconda创建、删除虚拟环境Linux
我这里就认为你已经安装了anaconda了，安装pytorch之前，我建议先创建一个虚拟环境，在虚拟环境中安装pytorch
首先查看自己的CUDA版本，教程参考Linux 系统查看CUDA和cuDNN版本号
我这里的CUDA版本为9.0.176

随后我们到Pytorch官网，根据自己的情况选择START LOCALLY，它就会自动生成你应该使用的安装命令。如果没有显卡CUDA一栏选择None就可以了。因为我的CUDA版本为9.0，所以我选择cudatoolkit=9.0，并指定了pytorch和torchvision的版本。这里的-c pytorch意思是选择从pytorch官方源下载，这里不推荐，官方源下载太慢了，还容易莫名其妙的中断，所以这里就去掉了，可看另一篇博客Anaconda换国内源。下载速度会有加成。

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0

OK，输入上面这个命令后我们等就可以了，中间回确认安装，输入y+ 回车就可以了。
下面这个就是安装成功了，接下来验证一下。

这里引入pytorch的包应输入import torch

# import这个包
import torch
# 这里验证 gpu 可用， True表示可用
torch.cuda.is_available()

妥了！

---

由于水平有限，文中错误之处在所难免，恳请大佬不吝赐教！