http://blog.csdn.net/coding_sun/article/details/78975486
登录PyTorch官网:http://pytorch.org/
查看本机的cuda版本
cat /usr/local/cuda/version.txt
利用conda(官网推荐用Anaconda安装)安装Linux版的的PyTorch,选择Python2.7, CUDA 8版本的。安装的时候就一条命令conda install pytorch torchvision -c pytorch,发现特别慢(墙的原因),所以参考这位大神的博客。
用清华TUNA镜像源,包含Anaconda仓库的镜像,将其加入conda的配置,配置如下,在终端分别输入以下两条命令:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
同时需要在~/.condarc中把- defaults那行删除,这样每次安装新的软件包就会从TUNA镜像下载。然后再输入安装命令即可(conda install pytorch torchvision cuda80 -c soumith)。选择用conda安装linux,Cuda版本:cuda8.0,Python版本:python2.7。
安装成功:
推荐PyTorch中文网:https://ptorch.com/