Ubuntu20.04安装CUDA及cudnn

CUDA安装

默认显卡驱动已经安装好了，使用nvidia-smi查看

安装对应cuda(cuda-11.0)
下载:https://developer.nvidia.com/cuda-11.0-download-archive
文件选择如图:

    # 下载链接以及安装命令
    wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run
    sudo sh cuda_11.0.2_450.51.05_linux.run

中间有部分选择，驱动已经有的，去掉勾选，然后移动到下方Install安装:

安装cudnn(cudnn v8.0.5)

下载: https://developer.nvidia.com/rdp/cudnn-archive
将下载的 cuDNN Library for Linux (x86_64) 解压，复制解压出来的文件到安装好的CUDA环境中

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo cp cuda/include/* /usr/local/cuda/include/
#更改权限输入 
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

编辑环境配置vim ~/.bashrc:

# CUDA Soft Link
export PATH=/usr/local/cuda-11.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64:/usr/local/cudnn/lib64:$LD_LIBRARY_PATH

激活环境source ~/.bashrc
完成之后就可以看到cuda信息

nvcc -V

显示如下：