[Python知识库] Ubuntu18.04安装驱动、cuda、cudnn（亲测有效）

开发: C++知识库 Java知识库 JavaScript Python PHP知识库人工智能区块链大数据移动开发嵌入式开发工具数据结构与算法开发测试游戏开发网络协议系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑笔记本显卡显示器固态硬盘硬盘耳机手机 iphone vivo oppo 小米华为单反装机图拉丁

-> Python知识库 -> Ubuntu18.04安装驱动、cuda、cudnn（亲测有效） -> 正文阅读

[Python知识库]Ubuntu18.04安装驱动、cuda、cudnn（亲测有效）

1、Ubuntu18.04安装显卡驱动

禁用SecureBoot（按照主板型号，查询开机进入bios按键，关闭secureboot）

禁用nouveau

sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
#在文件中插入以下内容，将nouveau加入黑名单，默认不开启
blacklist nouveau
options nouveau modeset=0
#输入以下命令使禁用生效然后重启
sudo update-initramfs -u
sudo reboot
#重启后验证（输入后没反应即为成功）
lsmod | grep nouveau

安装驱动

#1、安装前保证驱动卸载干净
sudo ./NVIDIA-Linux-x86_64-384.59.run(修改各自版本号) --uninstall
sudo apt-get remove --purge nvidia*
#2、ppa源驱动安装，首先查看电脑最适合的版本
ubuntu-drivers devices
#加入ppa源
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-driver-435 #此处数字要对应上面查询到的版本号
sudo apt-get install mesa-common-dev
#安装后重启即可，查看是否安装成功 
nvidia-smi

参考：Ubuntu18.04安装Nvidia显卡驱动教程_知遥君的博客-CSDN博客_ubuntu18.04安装nvidia显卡驱动

2、安装cuda

首先进入官网下载对应的cuda版本（注：下载cuda前，要搞清楚显卡的型号，深度学习代码要求的环境，如A6000显卡，使用pytorch只能安装cuda11.1版本以上的，否则运行代码时会提示算力不匹配）
选择runfile(local)方式下载安装，（速度快）

#（注：如显卡驱动已经安装，要取消重复安装显卡，也可以自动配置需要安装的路径，是否覆盖cuda软链接等）
sudo sh cuda_10.1.243_418.87.00_linux.run

添加环境变量

#在home目录的.bashrc中导入cuda的环境变量
export CUDA_HOME=/usr/local/cuda 
export PATH=$PATH:$CUDA_HOME/bin 
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
#退出.bachrc文件，刷新
source ~/.bashrc

输入nvcc -V查看是否安装成功

参考：Ubuntu18.04安装Cuda10.1/Cudnn_知遥君的博客-CSDN博客

3、安装cudnn

按照官网确定版本号（如A6000在cudnn8是能加快渲染速度）

下载，解压

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*