Ubuntu进行深度学习配置

驱动安装、CUDA安装和cuDNN

Posted by 周自横 on December 24, 2020

1.安装显卡驱动

首先卸载旧的显卡驱动

sudo apt-get purge nvidia*
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update  

添加官方ppa源并且刷新软件库

ubuntu-drivers devices

检测你的NVIDIA图形卡和推荐的驱动程序的模型

会显示有推荐安装程序,进行安装

sudo apt-get hold nvidia-driver-xxx

重启系统,检测

nvidia-smi

将会看到显卡信息

2. 安装CUDA

根据文档(https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html) 查看自己驱动对应的CUDA工具包版本,然后到网页(https://developer.nvidia.com/cuda-toolkit-archive) 下载自己的对应版本,本次选择的是deb(local)使用给的命令进行安装,减少设置,最后使用

nvcc -V

检查安装情况

3. 下载cuDNN

打开网页(https://developer.nvidia.com/zh-cn/cudnn) 下载cuDNN需要账户登录,安装对应的CUDA版本下载对用的,下载之后进行解压

tar -zxvf cudnn-XXXXXX

会解压到cuda文件夹中,然后按照对应子文件夹的归属,将文件复制到安装的CUDA驱动文件夹即/usr/local/cuda/

   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*

我在复制的时候直接复制了全部,直接使用了*,最后检查清理使用

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

image-20201223001048511

但是我并没有出现这个显示,但是深度学习程序还是直接测试成功了