ubuntu安装CUDA、cudnn

1.查看GPU

nvidia-smi

2.安装CUDA

https://developer.nvidia.com/cuda-downloads下载相应版本，需要先安装合适的GCC版本。

将CUDA添加到环境变量

sudo gedit ~/.bashrc 
export PATH=/usr/local/cuda-10.2/bin${PATH:+:$PATH}}    
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
source /etc/profile

3.安装cudnn

https://developer.nvidia.com/rdp/cudnn-archive下载相应版本，需要注册账号。

sudo dpkg -i <name>，其中<name>为刚才下载的deb文件名

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ 
sudo chmod a+r /usr/local/cuda/include/cudnn.h 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

验证查看CuDNN版本

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