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

二：安装CUDA(如果你不确定安装哪个版本，就去掉，默认最新)

conda install cudatoolkit=10.0 -n  momo -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

三：安装CUDNN

conda install cudnn=7.6.0 -n  momo  -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

四：安装pytorch（已经更换清华源，如果torchvision-0.4.1很慢，就ctrl+c关掉，在执行一遍本条命令）

 conda install pytorch torchvision cudatoolkit=10.0

最后赠送一个cv2的安装以及scipy安装，不要问为什么，问就是跑程序会报错

conda install -c menpo opencv

conda install scipy