Ubuntu18.04安装NVIDIA驱动与CUDA、CUDNN，亲测好用，百遍重装经验

最新推荐文章于 2024-07-11 18:55:55 发布

原创最新推荐文章于 2024-07-11 18:55:55 发布 · 988 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#ubuntu #cuda #nvidia

Ubuntu 专栏收录该内容

21 篇文章

订阅专栏

本指南详细介绍了在Ubuntu系统中安装CUDA和CUDNN的步骤，包括禁用nouveau驱动、下载安装包、设置环境变量及验证安装是否成功。

部署运行你感兴趣的模型镜像

让我们开始吧

准备工作

准备工作

1、开机进ubuntu设计界面按e禁用nouveau驱动操作流程
2、去nvidia官网下载CUDA安装包下载传送门
在这里插入图片描述
3、下载CUDNN我是传送门

4、安装gcc 6 或者降级降级

sudo apt-get install gcc-6

5、安装make

sudo apt-get install make

安装CUDA

1、给run安装包可执行权限

chmod 777 cudaxxx1.run #有多少个执行多少次
chmod 777 cudaxxx2.run

2、执行

sudo ./cudaxxx1.run #有多少个执行多少次
sudo ./cudaxxx2.run

安装时一般都选accept default y
只需注意若是已经安装高于cuda推荐版本的nvidia驱动时，在选择是否安装驱动是选n，反之y
3、设置环境变量

sudo gedit ~/.bashrc

在最后加入

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

4、验证cuda是否安装成功
重启终端输入

nvcc -V

出现如下
在这里插入图片描述

安装CUDNN

1、解压

tar -xvf cudnnxxx.tar

2、复制

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

出现如下，即成功
在这里插入图片描述

永久禁用nouveau

sudo gedit /etc/modprobe.d/blacklist.conf

输入密码后在最后一行加上: blacklist nouveau 是将Ubuntu自带的显卡驱动加入黑名单
Ctrl +S保存后注意此时还需执行以下命令使禁用 nouveau 真正生效终端输入：

sudo update-initramfs -u

重启电脑

附录

1、查看已装nvidia驱动的版本

sudo dpkg --list | grep nvidia-*

2、卸载nvidia驱动或者cuda

-------------nvidia
sudo apt-get --purge remove nvidia-*
# 也有的说用下面的命令进行卸载，我个人均有尝试，无碍
sudo apt-get --purge remove "*nvidia*"
-------------cuda
sudo apt autoremove cuda
# 若上一步删不干净，接着使用：
sudo apt --purge remove "*cublas*" "cuda*"