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

可能會有一些需要升級或安裝的包，一路升級安裝就行了。