Ubuntu 16.04创建虚拟环境并安装PyTorch GPU版

一、配置清华镜像，这样安装包的时候会快不少

在anaconda安装好之后，默认的镜像是官方的，由于官网的镜像在境外，使用国内的镜像能够加快访问的速度。这里选择了清华的的镜像。

在Anaconda Prompt中运行：

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 --set show_channel_urls yes

前往当前用户的目录下，查看.condarc文件：

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - defaults
show_channel_urls: true

 注意：配置完成之后，要关闭控制台再重新打开控制台，这样新配置的文件才会被加载进来

二、创建虚拟环境，注意创建虚拟环境时需要指定环境位置，否则可能不是创建在anaconda/envs下面（可能是配置了清华镜像的缘故）

conda create --prefix=/home/anaconda/envs/yourEnvironmentName python=3.6

三、激活虚拟环境

conda activate yourEnvironmentName

四、安装pytorch及CUDA和cudnn, 注意安装包时需要指定虚拟环境，否则可能会安装到别的虚拟环境中（可能是配置了清华镜像的缘故）

conda install pytorch torchvision cudatoolkit=10.1 cudnn=7.6.4 -n yourEnvironmentName

五、验证pytorch是否安装成功

进入python命令行，输入：

import torch
print(torch.cuda.is_available())

如果输出为True即代表安装成功