Ubuntu Nvidia GPU机器学习环境搭建(安装Nvidia Driver)

本文主要介绍为Ubuntu安装Nvidia GPU驱动,演示的环境的硬件为AWS EC2 g4dn机型,它搭载了一颗服务器专用的T4 Tensor Core GPU,这台服务器运行Ubuntu Server 22.04 LTS

Amazon EC2 G4dn 实例专用于加速机器学习推理和图形密集型工作负载,它搭载的T4 Tensor Core GPU有16GB的显存,非常适合中等负荷下的AI场景。本次我使用的是最经济的g4dn.xlarge。你自己也可以组装在你的个人电脑(需要Nvidia消费级显卡)上正常复现安装流程。

安装驱动

检查环境

输入uname -acat /etc/lsb-release进行操作系统检查,我们看到正在运行的是Ubuntu Server 22.04.3 LTS(内核为专为AWS云上环境优化后的Linux 6.2)。

安装依赖项

我们先安装一些依赖项,比如gcc,g++,make

我们执行:sudo apt update -y && sudo apt upgrade -y && sudo apt install vim gcc g++ make python3-pip -y

查询可用的Nvidia软件版本

我们可以运行nvidia-detectorubuntu-drivers devices查看可以使用的驱动版本。如果上述的2条命令找不到,可以输入sudo apt install ubuntu-drivers-common进行安装。

然后使用sudo apt install nvidia-driver-535 -y安装最新的535版本驱动(你可以安装nvidia-driver-535-server版本)

安装完成后需要重启服务器,我们直接sudo reboot就好

检测驱动安装

重启后,我们运行nvidia-smicat /proc/driver/nvidia/version进行检查。我们可以看到535版本的驱动安装好了

参考

相关推荐
要阿尔卑斯吗8 小时前
提示词优化启示:为什么“按顺序输出“比“关键度评分“更有效
后端
她的男孩9 小时前
后台接口加密别只会 HTTPS,ForgeAdmin 的 RSA + SM4/AES 源码拆解
后端·面试·开源
极光技术熊9 小时前
Spring AI 从入门到精通:构建你的 AI 开发知识体系
后端·github
程序员cxuan9 小时前
一句话,让你用上 GPT-5.6
人工智能·后端·程序员
远航_9 小时前
OpenSpec 完整详细介绍
前端·后端
AskHarries9 小时前
不用公网 IP,把 Windows 和 Linux 服务器放进同一个局域网:Tailscale 组网实战
后端
神奇小汤圆9 小时前
Java 的1 亿次对象创建:JVM 开启 / 关闭逃逸分析,GC 性能差距巨大
后端
tangdou3690986559 小时前
AI真好玩系列-2分钟快速了解DeepAgents | Quick Guide to DeepAgents in 2 Minutes
前端·javascript·后端
神奇小汤圆9 小时前
面试官:MySQL 为什么要是使用 MVCC?原理是什么?
后端
像我这样帅的人丶你还9 小时前
Java 后端详解(五):Redis 缓存
java·后端·全栈