当前位置:   article > 正文

AIGC | 为机器学习工作站安装NVIDIA 4070 Ti Super显卡驱动

AIGC | 为机器学习工作站安装NVIDIA 4070 Ti Super显卡驱动

8f93d7c3dbc564df06880970173ab1f6.gif

[ 知识是人生的灯塔,只有不断学习,才能照亮前行的道路 ]

0x00 前言简述

话接上篇《AIGC | Ubuntu24.04桌面版安装后必要配置》文章,作为作者进行机器学习的基础篇(筑基期),后续将主要介绍机器学习环境之如何在Ubuntu24.04桌面系统中进行NVIDIA显卡驱动安装,CUDA Toolkit安装,以及cuDNN的安装,以作者实践经历帮助读者快速搭建机器学习环境。

希望各位看友可以持续关注《#AIGC学习之路》学习更多AIGC相关文章,后续将陆续推出更多深度学习、强化学习、计算机视觉等系列专栏,敬请期待。

原文链接:AIGC | 为机器学习工作站安装NVIDIA 4070 Ti Super显卡驱动主要介绍机器学习环境之如何在 Ubuntu 24.04 TLS Desktop 系统中快速进行NVIDIA GeForce RTX 4070 Ti SUPER的驱动安装,并解决安装过程中的相关问题。icon-default.png?t=N7T8https://mp.weixin.qq.com/s/88Pa3dM_5jHHWv2dO285bA

0x01 基础依赖

1.安装显卡驱动

描述:通常如果你在Ubuntu24.04TLS桌面系统版本安装时,勾选【为图形和WIFI硬件安装第三方软件】后续就不用再安装显卡驱动,但是有可能安装版本和你所需要的学习实践环境版本不一致,所以有可能需要我们手动安装显卡驱动,这里作者也简单的实践一下在 Ubuntu 24.04 TLS Desktop 中进行NVIDIA GeForce RTX 4070 Ti SUPER的驱动安装,算是后来的朋友做一个参考。

0aedd60656c45a19ffe0d4e847792148.png

weiyigeek.top-GEFORCE RTX 4070图

温馨提示:若是你还没有安装Ubuntu 24.04桌面版本的童鞋,可以参考作者前面的《AIGC | 机器学习工作站安装Ubuntu24.04桌面版》文章进行系统安装与常规配置。

前置操作步骤:

Step 1.在终端中执行如下命令,安装必备软件,如果你是按照作者所写的《#AIGC学习之路》专栏文章安装的,那么你就可以跳过这一步。

  1. # 更新源
  2. sudo apt-get update
  3. # 安装必备软件(必须检查是否安装下述编译工具)
  4. sudo apt-get install g++ gcc make

Step 2.在安装NVIDIA驱动前,需要禁止系统自带显卡驱动nouveau,操作命令配置如下:

  1. ## 在终端输入命令打开 blacklist.conf 文件
  2. ## gedit/vim/vi 均可, 在文件末尾增加以下内容并保存
  3. sudo gedit /etc/modprobe.d/blacklist.conf
  4. blacklist nouveau
  5. # blacklist lbm-nouveau
  6. options nouveau modeset=0
  7. alias nouveau off
  8. alias lbm-nouveau off
  9. # 更新 initramfs 并重启电脑
  10. sudo update-initramfs -u
  11. sudo reboot
  12. # 再次检查 nouveau 是否被禁用如果没有输出则说明已禁用 nouveau
  13. lsmod | grep nouveau
  14. # 进入tty终端模式
  15. sudo telinit 3
  16. ## 如需重新打开图形界面
  17. # sudo telinit 5
  18. # 终端登录,输入用户名和密码
  19. # 关闭图形进程
  20. sudo service gdm3 stop
  21. sudo service lightdm stop

驱动安装步骤

Step 1.访问NVIDIA官网 [https://www.nvidia.com/Download/index.aspx?lang=cn], 选择你对应 NVIDIA 产品确定合适的驱动,并下载驱动安装包,例如,作者的 GEFORCE RTX 4070 显卡。

1d4428b620a659480cbb14903573954f.png

weiyigeek.top-下载驱动安装包图

Step 2.获取到驱动安装包后,在终端中执行如下命令进行安装。

  1. ## 进入驱动安装包所在目录
  2. cd /tmp
  3. wget https://cn.download.nvidia.com/XFree86/Linux-x86_64/550.90.07/NVIDIA-Linux-x86_64-550.90.07.run
  4. ## 赋予执行权限
  5. sudo chmod 777 NVIDIA-Linux-x86_64-550.90.07.run
  6. ## 安装:不安装OpenGL, 安装时关闭X服务 -no-x-check (可选)
  7. sudo ./NVIDIA-Linux-x86_64-535.54.03.run –no-opengl-files
  8. ## Install Nvidia's 32-bit compatibility libraries?
  9. ## 选择 "No"
  10. ## Would you like to run the nvidia-xconfig utility to automatically update your X configuration file so that the NVIDIA X driver dill be used dhen you restart X? Any pre-existing X configuration file will be backed up.
  11. ## 选择 "Yes"

Step 3.安装完成后,返回图形页面服务

  1. # 启动图形页面服务
  2. sudo service gdm3 start
  3. # 返回图形页面
  4. sudo telinit 5

Step 4.最后,在终端中执行如下nvidia-smi命令查看是否安装成功,以及查看驱动支持最高的CUDA版本。

  1. root@Super-Server:~# nvidia-smi
  2. Thu Jul  4 09:41:48 2024
  3. +-----------------------------------------------------------------------------------------+
  4. | NVIDIA-SMI 550.90.07              Driver Version: 550.90.07      CUDA Version: 12.4     |
  5. |-----------------------------------------+------------------------+----------------------+
  6. | GPU  Name                 Persistence-M | Bus-Id          Disp.A | Volatile Uncorr. ECC |
  7. | Fan  Temp   Perf          Pwr:Usage/Cap |           Memory-Usage | GPU-Util  Compute M. |
  8. |                                         |                        |               MIG M. |
  9. |=========================================+========================+======================|
  10. |   0  NVIDIA GeForce RTX 4070 ...    Off |   00000000:C1:00.0  On |                  N/A |
  11. |  0%   38C    P8              1W /  285W |      97MiB /  16376MiB |      0%      Default |
  12. |                                         |                        |                  N/A |
  13. +-----------------------------------------+------------------------+----------------------+
  14. +-----------------------------------------------------------------------------------------+
  15. | Processes:                                                                              |
  16. |  GPU   GI   CI        PID   Type   Process name                              GPU Memory |
  17. |        ID   ID                                                               Usage      |
  18. |=========================================================================================|
  19. |    0   N/A  N/A      2407      G   /usr/lib/xorg/Xorg                             77MiB |
  20. |    0   N/A  N/A      2673      G   /usr/bin/gnome-shell                           11MiB |
  21. +-----------------------------------------------------------------------------------------+

由上可知,驱动安装成功,显示驱动版本为 550.90.07,且支持最高CUDA 12.4版本。

后续我们将进行CUDA 12.4版本安装,请继续查看下一篇文章。

文章都看完了,都不关注一下作者吗?

温馨提示:作者最近10年的工作学习笔记(涉及网络、安全、运维、开发),需要学习实践笔记的看友,可添加作者账号[WeiyiGeeker],当前价格¥199,除了获得从业笔记的同时还可进行问题答疑以及每月远程技术支持,希望大家多多支持,收获定大于付出!

834203264e4e18b0026293f2f3848e6c.png


如果此篇文章对你有帮助,请你将它转发给更多的人!

 学习推荐 往期文章

推荐阅读
相关标签