ubuntu安装cuda及cudnn

2022/10/21 posts

# 准备

# 查看显卡驱动版本

nvidia-smi

image-20221021165542949

若未安装显卡驱动,则先去安装对应的显卡驱动

输出右上角的CUDA Version 即为可安装的最高CUDA版本

# 下载

去官网下载自己需要的版本 CUDA Toolkit - Free Tools and Training | NVIDIA Developer (opens new window)

下面教程以 11.3 为例

# 安装CUDA

sudo bash cuda_11.3.1_465.19.01_linux.run
  • 输入 accept 回车

    image-20221021170052555

  • 只勾选 CUDA Toolkit 即可

    image-20221021170147466

  • 安装完成

    image-20221021170224254

  • 配置环境变量

    • 编辑 ~/.bashrc 文件

      nano ~/.bashrc
      
    • 末尾添加以下内容

      export PATH=/usr/local/cuda-11.3/bin${PATH:+:${PATH}}
      export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
      
    • 使生效

      source ~/.bashrc
      
  • 验证安装

    nvcc -V
    

    出现下图内容即表示安装成功

    image-20221021170620323

# 安装CUDNN

  • 解压

    tar -xvf cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar.xz
    
  • 复制文件到cuda环境中

    • 进入解压后的文件夹

      cd cudnn-linux-x86_64-8.5.0.96_cuda11-archive/
      
    • 复制文件

      sudo cp include/cudnn.h  /usr/local/cuda/include
      sudo cp lib/libcudnn* /usr/local/cuda/lib64
      
    • 添加权限

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