ubuntu16.04下用miniconda安装pytorch GPU环境记录

1.安装miniconda

进入Miniconda官网Miniconda下载相应版本

我下载的是Python3.7 64bit版本
下载完成后，cd进入下载目录，执行

sh YOUR_FILE_NAME -b

如sh Miniconda3-latest-Linux-x86_64.sh -b

然后执行

~/miniconda3/bin/conda init

此时重新启动终端，执行

conda -V

如果不报错则安装成功

2.配置pytorch环境

首先新建一个环境，如环境名设为pytorch，python版本设为3.7

conda create -n pytorch python=3.7

环境创建完毕后，重启终端，执行

conda activate pytorch

接下来就是pytorch及CUDA的安装
由于通过Pytorch官网进行安装，网络会极慢。所以需要设置清华镜像源，进行满速下载。
依次执行如下命令：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ 
conda config --set show_channel_urls yes

接着开始正式安装：
首先进入pytorch官网选取合适的版本组合，如果该页面提供的版本都不合适，可进入pytorch历史版本查找合适的版本。
这里我选择了pytorch1.0.0+torchvision0.2.1+cuda8.0版本，并找到对应的如下安装命令：

conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 -c pytorch

但是此时还需要将指令末的pytorch替换为清华源，如下：

conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

然后执行它，片刻之后即可安装成功

3.验证安装是否成功

在该环境中，依次执行如下命令

python
import torch                     # 如正常则静默
a = torch.Tensor([1.])           # 如正常则静默
a.cuda()                         # 如正常则返回"tensor([ 1.], device='cuda:0')"
from torch.backends import cudnn # 如正常则静默
cudnn.is_acceptable(a.cuda())    # 如正常则返回 "True"

如果全部都未报错，那么就一切安装成功！