ubuntu 安装 cuda 和 cudnn

发表于 更新于 分类于 阅读次数： 本文字数： 1.1k 阅读时长 ≈ 1 分钟
记录安装cuda的步骤（还是archlinux好）。

一、CUDA

1. 下载 cuda

cuda 下载地址

或者直接终端下载：

1
2
wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
sudo sh cuda_10.1.243_418.87.00_linux.run

2. 安装

cuda8.0 后就集成了 nvidia 显卡驱动，cuda 安装可分为

  • 不带驱动安装
  • 带驱动安装

以下为不带驱动安装：

img
img

输入 accept

img
img

不选择 Driver

3. 设置 cuda 环境变量

sudo gedit ~/.bashrc

在里面添加：

1
2
3
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export PATH=$PATH:/usr/local/cuda-10.1/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1

== 目录要根据实际安装版本而定 ==

source ~/.bashrc

4. 检查是否安装成功

nvcc --version

img
img

二、CUDNN

  1. 下载 cudnn

    cudnn 下载地址

  2. 安装

    • deb 安装

      img
      img

      下载如图三个文件

      1
      2
      3
      sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb
      sudo dpkg -i libcudnn7-dev_7.0.3.11-1+cuda9.0_amd64.deb
      sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.deb

    • tgz 安装

      解压下载文件

      终端进入解压出来的文件目录

      1
      2
      3
      4
      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*

      （拷贝库和同文件并给予权限）

  3. 检查安装是否成功

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

参考文章

https://blog.csdn.net/sss_369/article/details/94591280

https://blog.csdn.net/sss_369/article/details/94592268