机器学习环境搭建 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版本的机器学习环境搭建部分,流程基本是一样的。

相关推荐
你的人类朋友19 小时前
说说签名与验签
后端
databook19 小时前
Manim实现脉冲闪烁特效
后端·python·动效
canonical_entropy1 天前
AI时代,我们还需要低代码吗?—— 一场关于模型、演化与软件未来的深度问答
后端·低代码·aigc
颜如玉1 天前
HikariCP:Dead code elimination优化
后端·性能优化·源码
考虑考虑1 天前
Jpa使用union all
java·spring boot·后端
bobz9651 天前
virtio vs vfio
后端
Rexi1 天前
“Controller→Service→DAO”三层架构
后端
bobz9651 天前
计算虚拟化的设计
后端
深圳蔓延科技1 天前
Kafka的高性能之路
后端·kafka
Barcke1 天前
深入浅出 Spring WebFlux:从核心原理到深度实战
后端