0
点赞
收藏
分享

微信扫一扫

【Linux】Ubuntu 20.04 安装NVIDIA驱动 + CUDA + CUDNN

Alex富贵 2022-03-12 阅读 154

1. 安装NVIDIA驱动

首先检查系统推荐安装的版本

ubuntu-drivers devices

安装推荐版本驱动

sudo apt install nvidia-driver-XXX

安装完毕后重启,检查

nvidia-smi

2. 安装CUDA

从NVIDIA官网选择合适版本下载

推荐使用runfile安装

在安装过程中选择Continue,输入accept,使用空格去掉Driver,最后install安装

安装完毕之后修改路径文件

gedit ~/.bashrc

在最后加上一下两句,注意修改CUDA版本

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

保存路径

source ~/.bashrc

查看CUDA版本以验证安装成功

nvcc -V

3. 安装CUDNN

从官网中下载合适版本,需要注册登录NVIDIA账号

下载完之后,解压文件夹,并复制文件(注意修改路径):

将cuda/include/cudnn.h文件复制到usr/local/cuda/include文件夹

sudo cp path/cuda/include/cudnn.h /usr/local/cuda/include

将cuda/lib64/下所有文件复制到/usr/local/cuda/lib64文件夹

sudo cp path/cuda/lib64/* /usr/local/cuda/lib64

复制完毕之后,添加读取权限

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

验证安装成功:

首先查看CUDNN版本文件路径

cat /usr/local/cuda/include/cudnn.h | grep cudnn
find / -name cudnn_version.h 2>&1 | grep -v "Permission denied"

使用以下命令查看CUDNN版本(注意修改文件路径)

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

感谢以下作者:

@ LUKE [链接]
@ ashome123 [链接]
@ 哈希Ma [链接]
@ eaxy_z [链接]

举报

相关推荐

0 条评论