Ubuntu18.04 +anconda3.6+cuda +cudnn +Pytorch-gpu环境搭建

1、安装anconda
可以安装的方式（直接在linux中下载，下载后传输到linux中），我是官网下载完上传安装的，因为网速真的很差
下载xxx.sh文件上传到linux中，我上传的是/home路径下
执行：1、bash Anaconda3-5.1.0-Linux-x86_64.sh
按步骤运行即可，最后有个是否安装VS的yes or no ，输入no即可
安装完了
和win环境一样，也得配置环境变量：
打开文件：vim ~/.bashrc。
在文件最后加入：export PATH=$PATH:/root/anaconda3/bin
保存退出即可
更新：source ~/.bashrc
输入：python,输出python版本信息即成功
2、创建虚拟环境
conda create -n envy-tensorflow python=3.6
启动虚拟环境指令
source activate envy-pytorch
3、安装cuda、cudnn、pytorch
在conda虚拟环境执行:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
conda list：查看虚拟环境是否成功安装的软件
测试pytorch是否可以调用gpu:
import torch
flag = torch.cuda.is_available()
print(flag)
输出：True，成功