ubuntu安装cuda和cudnn

ubuntu安装cuda(10.0)和cudnn

  • 安装显卡驱动
    • cuda和graphic driver版本对应关系
  • 安装CUDA10.0
    • 下载cuda
    • 安装cuda
    • 查看安装的cuda版本
    • 检测是否安装成功
  • 安装cudnn
    • 下载cudnn
    • 安装cudnn
    • 查询cudnn版本

安装显卡驱动

cuda和graphic driver版本对应关系

一般根据需要,安装指定的cuda版本,因此显卡驱动的版本需要满足最低的版本要求。

ubuntu安装cuda和cudnn_第1张图片

查看可安装的驱动版本

ubuntu-drivers devices

自动安装推荐的驱动版本

如果推荐版本满足要求,可以自动安装推荐版本的驱动。

sudo ubuntu-drivers autoinstall

安装后重启

检查驱动是否安装成功

nvidia-smi

显示最高可安装的cuda版本为10.2,当前驱动版本为440。

ubuntu安装cuda和cudnn_第2张图片

安装CUDA10.0

下载cuda

参考cuda安装方法,进入cuda官方下载链接下载安装即可;

安装cuda

由于已经有了显卡驱动,不需要进行安装;
cuda安装在默认路径即可;

sudo sh cuda_10.0.130_410.48_linux.run

ubuntu安装cuda和cudnn_第3张图片

查看安装的cuda版本

cat /usr/local/cuda-10.0/version.txt

在这里插入图片描述

检测是否安装成功

nvcc -V

在这里插入图片描述

安装cudnn

cudnn是GPU加速计算深层神经网络的库。

下载cudnn

参考cudnn安装方法,进入cudnn官方下载链接;
选择cuda10对应的cudnn版本,点击cudnn library for linux进行下载,注意需要注册登陆才能下载;

ubuntu安装cuda和cudnn_第4张图片

安装cudnn

解压

tar -xvf cudnn-10.0-linux-x64-v7.6.5.32.tgz

依次执行命令

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

查询cudnn版本

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

ubuntu安装cuda和cudnn_第5张图片
cudnn安装成功。

[1]:Ubuntu18.04 安装Cuda 10.0 和Cudnn
[2]:Ubuntu18.04安装cuda(10.1及10.0)和cudnn
[3]:cuda toolkit documentation
[4]:cuda官方下载链接
[5]:cudnn官方下载链接

你可能感兴趣的:(ubuntu,深度学习,cuda)