linux下通過anaconda安裝pytorch

前提是已經安裝好anaconda了

1.首先根據cuda的版本來安裝正確的pytorch命令

cuda7.5：conda install pytorch torchvision cuda75 -c pytorch

cuda8.0:  conda install pytorch torchvision -c pytorch

cuda9.0：conda install pytorch torchvision cuda90 -c pytorch

這樣會發現安裝的很慢

因此加上如下命令：conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

若向安裝特定版本的pytorch，例如0.4.0，cuda8.0的pytorch則運行如下命令：

conda install pytorch=0.4.0 torchvision -c pytorch

注意 -c pytorch去掉，會發現安裝速度很快。

2.測試是否安裝成功

如何查看當前pytorch版本:進入python環境
import torch
print(torch.__version__)

安裝成功。