Ubuntu20.04安装nvidia显卡驱动/CUDA/CUDNN

本文目录
一、NVIDIA显卡驱动安装方式有哪些?
二、安装NVIDIA
三、安装CUDA
四、安装CUDNN
五、常见问题集锦

微信公众号:小贾探AI,不定期更新发布人工智能相关的技术文章

一、NVIDIA显卡驱动安装方式有哪些?

安装方式有三种:

  • ubuntu附加驱动的方式
  • 命令行方式安装
  • .run文件的方式进行安装。
1.ubuntu附加驱动的方式

点击菜单中的Additional Drivers选择适合的驱动版本进行安装,该方法最方便快捷(但有时会翻车)

2.命令行方式安装

更新所有的软件包

sudo add-apt-repository ppa:graphics-drivers/ppa  # 加入官方ppa源  
sudo apt update  # 检查软件包更新列表  
apt list --upgradable  # 查看可更新的软件包列表  
sudo apt upgrade  # 更新所有可更新的软件包  

安装显卡驱动

apt install ubuntu-drivers-common  
ubuntu-drivers devices    # ubuntu检测n卡的可选驱动(选择下面高亮带recommend)  
sudo apt install nvidia-driver-530  # 根据自己的n卡可选驱动下载显卡驱动  
完成后输入nvidia-smi验证。 
3 .run文件方式安装【我选这种方式】

该方法操作步骤相对比较麻烦,但最不容易出问题。

二、安装NVIDIA

1.查看操作系统
lsb_release -d
Ubuntu 20.04.1 LTS
cudnnGetVersion() : 8901 , CUDNN_VERSION from cudnn.h : 8901 (8.9.1)
2.查看显卡型号
root@localhost:/data/software# lspci | grep VGA 或 sudo lshw -numeric  -C display
03:00.0 VGA compatible controller: ASPEED Technology, Inc. ASPEED Graphics Family (rev 51)
31:00.0 VGA compatible controller: NVIDIA Corporation Device 2230 (rev a1)
4b:00.0 VGA compatible controller: NVIDIA Corporation Device 2230 (rev a1)
b1:00.0 VGA compatible controller: NVIDIA Corporation Device 2230 (rev a1)
ca:00.0 VGA compatible controller: NVIDIA Corporation Device 2230 (rev a1)
http://pci-ids.ucw.cz/mods/PC/10de?action=help?help=pci
输入上面的 2230 得出当前显卡信息为:Name: GA102GL [RTX A6000]
3.安装GCC、G++、make等工具
sudo apt-get install build-essential
gcc --version 验证gcc是否安装成功

sudo apt-get install g++

sudo apt-get install make
gcc (Ubuntu 9.4.0-1ubuntu1~20.04.1) 9.4.0
g++ is already the newest version (4:9.3.0-1ubuntu2)
make is already the newest version (4.2.1-1.2)
4.卸载nvidia旧驱动
sudo /usr/bin/nvidia-uninstall
sudo apt-get --purge remove nvidia*
sudo apt-get purge nvidia*
sudo apt-get purge libnvidia*
直到命令不输出任何内容:sudo dpkg --list | grep nvidia-*
5.根据显卡型号,下载对应的显卡驱动
  • 下载驱动网址:Official Drivers | NVIDIA
  • 可以选择版本下载,点击 Official Advanced Driver Search | NVIDIA
  • 根据自己的显卡型号,查找驱动,我这里选择NVIDIA-Linux-x86_64-525.116.04.run
  • 安装驱动之前,需要禁用nouveau
    (nouveau 是专门用来逆向工程 NVIDIA 闭源驱动的开源驱动项目),因为ubuntu安装时会默认安装nouveau驱动,如果不禁用,会阻碍NVIDIA官方驱动的安装。
  • 禁用nouveau
sudo vim /etc/modprobe.d/blacklist.conf
在文件末尾添加:
blacklist nouveau
blacklist lbm‐nouveau
options nouveau modeset=0
alias nouveau off
alias lbm‐nouveau off
对刚才修改的文件进行更新:sudo update-initramfs -u
重启计算机:sudo reboot
查看nouveau是否禁用成功:lsmod | grep nouveau,执行完这句,如没有任何输出,表示禁用成功。
6.关闭图形界面
(方式1)进入 tty模式
ctrl + alt+ F1 或 ctrl + alt+ F2 或 ctrl + alt+ F3,视情况
输入:sudo service lightdm stop
(方式2)进入纯命令模式,有图形化界面时不可以进行安装
init 3
授可执行权限安装文件  
sudo chmod +x NVIDIA-Linux-x86_64-384.59.run
运行安装文件   
sudo ./NVIDIA-Linux-x86_64-384.59.run -no-x-check -no-nouveau-check -no-opengl-files  
出现Extraction failed的错误:/tmp存储不够,需要自行修改脚本中的路径。
7.进入驱动所在目录,给驱动文件付权限,然后安装
sudo chmod +x NVIDIA-Linux-x86_64-510.68.02.run
sudo sh NVIDIA-Linux-x86_64-510.68.02.run  -no-opengl-files -no-x-check -no-nouveau-check
参数说明:
-no-x-check:安装驱动时关闭 X 服务
-no-nouveau-check:安装驱动时禁用 nouveau
-no-opengl-files:只安装驱动文件,不安装 OpenGL 文件

Would you like to register the kernel module souces with DKMS? This will allow DKMS to automatically build a new module, if you install a different kernel later?  选择 No 继续。
Nvidia's 32-bit compatibility libraries? 选择 No 继续。
Would you like to run the nvidia-xconfigutility to automatically update your x configuration so that the NVIDIA x driver will be used when you restart x? Any pre-existing x confile will be backed up.  选择 Yes  继续
8.挂载 Nvidia 驱动
sudo modprobe nvidia
9.查看驱动是否安装成功
nvidia-smi

三、安装CUDA

1、查看自己电脑的驱动程序版本

CUDA 12.1 Update 1 >=530.30.02
nvidia-smi可以看到最高版本的cuda不能大于12.1
具体驱动程序对应的cuda版本可以通过官网查看

2、下载cuda

官网地址:CUDA Toolkit Archive | NVIDIA Developer

因为我的驱动是510.47.03,所以我下载 CUDA Toolkit 11.8

卸载旧版本CUDA-12.1

sudo apt-get --purge remove "*cuda*" "*cublas*" "*cufft*" "*cufile*" "*curand*"  "*cusolver*" "*cusparse*" "*gds-tools*" "*npp*" "*nvjpeg*" "nsight*" "*nvvm*"
sudo apt-get autoremove
或
sudo /usr/local/cuda-12.1/bin/cuda-uninstaller
用空格选择所有cuda相关文件,确认,需要一会儿提示成功卸载。
最后删除磁盘文件:sudo rm -rf /usr/local/cuda-12.1

下载run脚本

wget https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run
vi cuda_11.8.0_520.61.05_linux.run 编辑修改/tmp为/data/tmp

执行安装cuda

sudo sh cuda_11.6.2_510.47.03_linux.run

配置环境变量

sudo  vim /etc/profile
在文件末尾添加:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64
export PATH=/usr/local/cuda-11.8/bin:$PATH
source /etc/profile
3、看看cuda安装是否正常:
root@localhost:/data/software# nvcc  -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Tue_Feb__7_19:32:13_PST_2023
Cuda compilation tools, release 12.1, V12.1.66
Build cuda_12.1.r12.1/compiler.32415258_0

cuda环境操作命令

  • 查看已有的环境:conda info --envs
  • 新建环境,并指定该环境的python版本:conda create -n xxxxx python=3.9
  • 激活环境:conda activate xxxxx
  • 退出环境:conda deactivate
  • 删除环境:conda remove -n xxxxx --all

四、安装CUDNN

1.下载cudnn(需要登录)

https://developer.nvidia.com/rdp/cudnn-download

2.安装cudnn
  • 点击这里查看安装说明Installation Guide: NVIDIA Deep Learning cuDNN Documentation

  • 安装步骤:
    下载Debian本地存储库安装包。在发出以下命令之前,必须替换X.Y和x.x.x与您特定的CUDA和cuDNN版本。
    【我的是8.5.0.96-1+cuda11.6】

首先解压并安装deb包

sudo dpkg -i cudnn-local-repo-ubuntu2004-8.5.0.96_1.0-1_amd64.deb

安装完后,它会自动提示把CUDA GPG Key导入的

$ sudo cp /var/cudnn-local-repo-ubuntu2004-8.5.0.96/cudnn-local-B0FE0A41-keyring.gpg /usr/share/keyrings/

更新一下软件库

$ sudo apt-get update

最后是真正开始安装cuDNN的库:

$ sudo apt-get install libcudnn8=8.5.0.96-1+cuda11.7
$ sudo apt-get install libcudnn8-dev=8.5.0.96-1+cuda11.7
$ sudo apt-get install libcudnn8-samples=8.5.0.96-1+cuda11.7

检查是否安装成功:
为了验证cuDNN已经安装并正常运行,编译位于Debian文件/usr/src/cudnn_samples_v8目录下的mnistCUDNN示例。

$ cp -r /usr/src/cudnn_samples_v8/ $HOME
$ cd  $HOME/cudnn_samples_v8/mnistCUDNN
$ make clean && make

如果在运行make指令的时候,出现了不能编译,提示缺少FreeLmage.h与资源库的问题
则运行下面指令安装FreeLmage相关文件:

$ sudo apt-get install libfreeimage3 libfreeimage-dev
$ ./mnistCUDNN

如果cuDNN在Linux系统上正确安装并运行,您将看到类似以下的消息:
测试通过!

  • 24
    点赞
  • 173
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
安装CUDAcuDNN的步骤如下: 1. 首先,下载CUDA安装文件。可以从NVIDIA官方网站下载对应版本的CUDA安装文件。使用wget命令下载安装文件,并将其移动到指定位置: ``` wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600 wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda-repo-ubuntu2004-11-2-local_11.2.0-460.27.04-1_amd64.deb sudo dpkg -i cuda-repo-ubuntu2004-11-2-local_11.2.0-460.27.04-1_amd64.deb sudo apt-key add /var/cuda-repo-ubuntu2004-11-2-local/7fa2af80.pub sudo apt-get update sudo apt-get -y install cuda ``` [1] 2. 接下来,安装显卡驱动。在安装CUDA之前,需要先卸载所有的驱动,并更新软件列表和安装必要的软件和依赖。查询硬件(显卡)信息,并从NVIDIA官网下载对应的驱动。最后,禁用默认驱动(nouveau是通用的驱动程序)。 [2] 3. 安装cuDNN。首先,从NVIDIA官方网站下载对应版本的cuDNN库文件。然后,将相关的库文件拷贝到CUDA安装目录中: ``` sudo cp include/cudnn* /usr/local/cuda/include/ sudo cp 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] 完成以上步骤后,您的Ubuntu 20.04系统就安装好了CUDAcuDNN。请确保按照上述步骤进行操作,并根据您的系统和硬件配置选择正确的版本。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值