Linux下用anaconda安装pytorch

最近跑实验有用到pytorch
我的anaconda是已经安装好了,也安装了tensorflow
现在用conda命令安装一下pytorch

首先创建虚拟环境

为pytorch创建一个虚拟环境，这是推荐的做法，这样在已有其他环境如tensorflow时，就不会互相冲突。在终端输入：

conda create -n pytorch python=3.5

这就创建了一个名为pytorch，python版本为3.5的虚拟环境。于是我们可以把Pytorch安装在这个环境下面。
首先激活这个环境：source activate pytorch

source activate pytorch

在Pytorch的官网，根据自己的机器配置选择对应的安装命令。
我的是

在终端输入：

conda install pytorch torchvision cudatoolkit=8.0 -c pytorch

可能会有一些需要升级或安装的包，一路升级安装就行了。