机器学习环境搭建 Debian 12 安装CUDA

提前安装依赖组件

检查Linux服务器上的VGA显卡信息,我们可以看到Nvidia GPU的具体型号

bash 复制代码
lspci -nn | egrep -i "3d|display|vga"

修改/etc/apt/sources.list,确保添加了contribnon-freenon-free-firmware组件

bash 复制代码
deb http://deb.debian.org/debian/ bookworm main contrib non-free non-free-firmware

先更新Linux并安装必要的编译组件

bash 复制代码
sudo apt update -y && sudo apt upgrade -y && sudo apt install vim gcc g++ make python3-pip -y

安装必要的nvidia gpu驱动依赖组件

bash 复制代码
sudo apt install  -y
sudo apt install linux-headers-amd64 linux-headers-$(uname -r) build-essential  -y

安装CUDA 12.3.1

打开CUDA Toolkit 12.3.1,选择Debian 12 X86_64版本并进行本地deb安装。

依次安装如下的命令

bash 复制代码
wget https://developer.download.nvidia.com/compute/cuda/12.3.1/local_installers/cuda-repo-debian12-12-3-local_12.3.1-545.23.08-1_amd64.deb
sudo dpkg -i cuda-repo-debian12-12-3-local_12.3.1-545.23.08-1_amd64.deb
sudo cp /var/cuda-repo-debian12-12-3-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt install software-properties-common
sudo add-apt-repository contrib
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-3
bash 复制代码
sudo apt-get install -y cuda-drivers
sudo apt install libglvnd-dev pkg-config firmware-misc-nonfree nvidia-kernel-dkms -y

安装完后进行检查

bash 复制代码
nvidia-smi

这里我们就完成了CUDA的安装,关于cuDNN和TensorRT可以参考 Ubuntu版本的机器学习环境搭建部分,流程基本是一样的。

相关推荐
武子康2 小时前
大数据-139 ClickHouse MergeTree 最佳实践:Replacing 去重、Summing 求和、分区设计与物化视图替代方案
大数据·后端·nosql
该用户已不存在2 小时前
7个让全栈开发效率起飞的 Bun 工作流
前端·javascript·后端
清空mega2 小时前
从零开始搭建 flask 博客实验(2)
后端·python·flask
G_dou_3 小时前
Rust安装
开发语言·后端·rust
9ilk3 小时前
【仿RabbitMQ的发布订阅式消息队列】--- 模块设计与划分
c++·笔记·分布式·后端·中间件·rabbitmq
一枚码仔3 小时前
SpringBoot启动时执行自定义内容的5种方法
java·spring boot·后端
桦说编程3 小时前
如何在Java中实现支持随机访问的固定窗口队列
java·数据结构·后端
oak隔壁找我4 小时前
Spring框架中的跨域CORS配置详解
java·后端
想不明白的过度思考者5 小时前
Rust——或模式(Or Patterns)的语法:Rust模式匹配的优雅演进
开发语言·后端·rust·模式匹配
JimmtButler5 小时前
RocketMQ本地编译
后端·rocketmq