ubuntu下如何安装pytorch,cuda,cudnn

一:安装anaconda

Linux下安装Anaconda，pytorch-GPU版本（请确定你的机器上是否有英伟达显卡）
首先：查看cuda的版本：cat /usr/local/cuda/version.txt

去官网下载对应版本的Anaconda，这里我要用的是python 3.6，因此我下载的安装包为：Anaconda3-5.3.1-Linux-x86_64.sh

1.新建环境

conda create -n momo python=3.6

2.进入环境

3.更换清华源

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