Ubuntu 16.04 安装gpu版本的pytorch

1.查看cuda和cudnn的版本

cat /usr/local/cuda/version.txt

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

2.安装好anaconda

3.配置anaconda的环境变量（如果输入conda提示：未找到命令，则需要这一步，配置环境变量）

4.更换源

conda config --add channels https://repo.continuum.io/pkgs/free/ 
conda config --add channels https://repo.continuum.io/pkgs/main

conda config --set show_channel_urls yes

5.创建conda环境

conda create -n pytorch（环境名）python=3.5

6.进入创建好的环境

conda activate pytorch（环境名）

7.安装cuda对应版本的torch

conda install pytorch torchvision cuda90 -c pytorch

   查看版本到pytorch官网或者https://blog.csdn.net/qq_41149269/article/details/81267591