Ubuntu安装CUDA和cuDNN

一、本文使用的环境

系统：Ubuntu 18.04.4

CUDA：cuda_11.0.2_450.51.05_linux

cuDNN：cudnn-11.1-linux-x64-v8.0.5.39

显卡驱动版本：450.80.02

二、安装CUDA

1.下载：CUDA Toolkit 11.0 Download | NVIDIA Developer

2.执行cuda_11.0.2_450.51.05_linux.run，注意不安装驱动，假设安装到默认目录/usr/local

3. vi ~/.bashrc,在最下面加上如下内容

export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64
export PATH=$PATH:/usr/local/cuda-11.0/bin:/usr/local/cuda-11.0/lib64:/usr/local/cuda-11.1:/usr/local/cuda-11.0/cuda/include:/usr/local/cuda-11.0/lib64:
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.0

4.执行 source  ~/.bashrc

三、安装cuDNN

1.下载 https://developer.nvidia.com/rdp/cudnn-download

2.下载到/usr/local/, 然后tar zxvf cudnn-11.1-linux-x64-v8.0.5.39.tgz

3.执行下面的命令查看是否安装成功：

nvcc --version
# 或者nvcc -V

  输出信息，表示安装成功：

四、安装显卡驱动

1.下载： 官方驱动 | NVIDIA， 选择自己的显卡型号

2.执行下载的文件就行

3. 执行nvidia-smi出现下面的内容，成功