【环境配置】Ubuntu20.04 安装cuda和cudnn

安装cuda

  1. https://developer.nvidia.com/cuda-toolkit-archive下载对应版本的cuda

  2. 进入到含有安装包的目录下,执行

    sudo sh cuda_10.2.89_440.33.01_linux.run
    
  3. 等待一会,出来下面的界面,输入accept
    【环境配置】Ubuntu20.04 安装cuda和cudnn
    4.cuda安装包是自带显卡驱动的,所以这一步按空格去掉安装显卡驱动的选项,然后选择install
    【环境配置】Ubuntu20.04 安装cuda和cudnn

  4. 安装完后,应该显示下面的界面
    【环境配置】Ubuntu20.04 安装cuda和cudnn

  5. 需要我们配置环境变量,接下来打开终端输入gedit ~/.bashrc,在文件最后输入

    export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
    

    注意:local后面的名字要安装自己下载的cuda版本,我这里是10.1,所以是cuda-10.1

  6. 更新环境变量
    source ~/.bashrc

  7. 验证,打开终端输入nvcc -V,显示如下则证明cuda安装成功

    nvcc: NVIDIA (R) Cuda compiler driver
    Copyright (c) 2005-2019 NVIDIA Corporation
    Built on Fri_Feb__8_19:08:17_PST_2019
    Cuda compilation tools, release 10.1, V10.1.105
    

安装cudnn

  1. https://developer.nvidia.com/cudnn下载相应版本的cudnn,需要登陆。

  2. 下载解压

    	tar -xvf cudnn-10.2-linux-x64-v7.6.5.32.tgz 
    
  3. cuda/include/cudnn.h文件复制到usr/local/cuda/include文件夹;
    cuda/lib64/下所有文件复制到/usr/local/cuda/lib64文件夹中,并添加读取权限:

    sudo cp ./cuda/lib64/* /usr/local/cuda/lib64
    sudo cp ./cuda/include/cudnn.h /usr/local/cuda/include
    sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
    
  4. 查看是否安装成功

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

    或者

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

    根据cudnn版本不同,使用的命令不同。

    显示如下:

    #define CUDNN_MAJOR 8
    #define CUDNN_MINOR 0
    #define CUDNN_PATCHLEVEL 4
    --
    #define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
    
    #endif /* CUDNN_VERSION_H */
    

    如果找不到的话,查看cudnn_version.h文件是否没有移动到/usr/local/cuda/include/下,新版本好像单独分离了,移动过去就可以了
    sudo cp ./cuda/include/cudnn_version.h /usr/local/cuda/include

上一篇:android 按两次返回键退出


下一篇:基于Python的图像超分辨率(Image Super Resolution)