Linux怎么配置系统的多显卡并行计算环境

阿强姑娘_5083

阿强姑娘_5083

2026-06-10

631人浏览

原创

能跑起来的核心是“认得全、分得清、不抢资源”:nvidia-smi需列出所有gpu,cuda_visible_devices要精准指定可见设备,xorg.conf须避免xorg占用计算卡,三者缺一不可。

linux怎么配置系统的多显卡并行计算环境

多显卡并行计算环境能不能跑起来,核心不在“配”,而在“认得全、分得清、不抢资源”——nvidia-smi 能列出所有卡,CUDA_VISIBLE_DEVICES 能精准指定,xorg.conf 不抢计算卡,这三件事没做对,后面全是空谈。

确认系统真正识别全部 GPU

别信硬件插了就等于系统看见了。很多问题从第一步就错了:你以为插了两张 RTX 4090,nvidia-smi 却只显示一张,或者 nvidia-smi -L 输出的 UUID 少一个。

  • 执行 nvidia-smi -L,每条输出对应一张物理卡;若数量不符,先查 lspci | grep -i nvidia 看 PCIe 设备是否全被枚举出来
  • 常见漏卡原因:BIOS 中禁用了 “Above 4G Decoding” 或 “Resizable BAR”,或某张卡 PCIe 插槽供电不足(尤其 4090 这类卡需双 8pin)
  • 如果某张卡在 nvidia-smi 里温度/功耗/显存占用始终为 0,大概率是驱动未加载其内核模块,可查 dmesg | grep -i nvidia 是否有 probe failed 或 bus error

避免 Xorg 占用计算用 GPU

这是 Linux 多显卡环境下最隐蔽的性能杀手:X Window 默认可能把高性能独显当显示输出设备绑定,导致你的训练脚本调用 cudaSetDevice() 时直接失败,报错 CUDA_ERROR_INVALID_DEVICE 或 device is busy。

银河易创
银河易创

一款集AI对话、绘画、音乐、视频和PPT等能力于一体的一站式AIGC创作平台。

下载
  • 用 lspci -k | grep -A 3 -i vga 查每张卡的 Kernel driver in use,确认哪张被 nvidia 驱动接管、哪张被 i915 或 amdgpu 管理
  • 编辑 /etc/X11/xorg.conf,明确指定仅用集成显卡(如 Intel)或低功耗独显负责显示,关键段落加 Option "AllowEmptyInitialConfiguration" "true" 和 Option "UseDisplayDevice" "none" 给计算卡
  • 更稳妥的做法:把计算卡从 Xorg 的 Device Section 中完全排除,只保留显示卡的配置;或改用 OutputClass 规则放在 /usr/share/X11/xorg.conf.d/ 下按 BusID 匹配

设置 CUDA_VISIBLE_DEVICES 控制可见性

这个环境变量不是“告诉 Ollama 或 PyTorch 用哪些卡”,而是“告诉 CUDA runtime 当前进程能看到哪些物理设备编号”。它直接影响 torch.cuda.device_count() 返回值和默认 device 选择。

  • 假设 nvidia-smi -L 输出顺序是:GPU 0: NVIDIA RTX 4090、GPU 1: NVIDIA RTX 4090、GPU 2: NVIDIA RTX 4090、GPU 3: NVIDIA RTX 4090,那么 CUDA_VISIBLE_DEVICES=0,2 后,进程内看到的设备编号会重映射为 0→原0、1→原2
  • 务必在启动命令前导出,例如:CUDA_VISIBLE_DEVICES=0,1 python train.py;写进 ~/.bashrc 会影响所有进程,不推荐
  • 注意:该变量对 nvtop、nvidia-smi 本身无效,它们永远显示物理视图;但对 torch.cuda.is_available() 和 cudaGetDeviceCount() 生效

驱动与 CUDA 版本必须严格匹配

驱动版本决定了你能用的最高 CUDA Toolkit 版本,而实际安装的 CUDA Toolkit 版本又决定 PyTorch / Ollama 等二进制包能否加载。三者错一个,import torch 就可能抛 libcudart.so not found 或 version mismatch。

  • 查当前驱动支持的 CUDA 最高版本:运行 nvidia-smi,右上角显示的 “CUDA Version: 12.4” 是驱动能兼容的上限,不是已安装版本
  • 查已安装 CUDA:看 /usr/local/cuda-xx 目录是否存在,再运行 /usr/local/cuda-12.1/bin/nvcc --version
  • PyTorch 官网下载链接里明确标着 “cu121” 或 “cu118”,必须和你 nvcc 输出的版本一致;Ollama 的 ollama run llama3 若提示 no CUDA-capable device,八成是 CUDA runtime 找不到对应驱动

真正麻烦的从来不是配置步骤,而是当你以为“已经配好”时,nvidia-smi 显示四张卡都在 idle,但 python -c "import torch; print(torch.cuda.device_count())" 却返回 0 —— 那说明 Xorg 还锁着其中一张,或者 CUDA_VISIBLE_DEVICES 被某个父 shell 意外覆盖了。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

linux显卡

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
磁盘配额是什么
磁盘配额是什么

磁盘配额是计算机中指定磁盘的储存限制,就是管理员可以为用户所能使用的磁盘空间进行配额限制,每一用户只能使用最大配额范围内的磁盘空间。php中文网为大家提供各种磁盘配额相关的内容,教程,供大家免费下载安装。

2023.06.21

3850

4

如何安装LINUX
如何安装LINUX

本站专题提供如何安装LINUX的相关教程文章,还有相关的下载、课程,大家可以免费体验。

2023.06.29

3313

10

linux find
linux find

find是linux命令,它将档案系统内符合 expression 的档案列出来。可以指要档案的名称、类别、时间、大小、权限等不同资讯的组合,只有完全相符的才会被列出来。find根据下列规则判断 path 和 expression,在命令列上第一个 - ( ) , ! 之前的部分为 path,之后的是 expression。还有指DOS 命令 find,Excel 函数 find等。本站专题提供linux find相关教程文章,还有相关

2023.06.30

2453

5

linux修改文件名
linux修改文件名

本专题为大家提供linux修改文件名相关的文章,这些文章可以帮助用户快速轻松地完成文件名的修改工作,大家可以免费体验。

2023.07.05

5890

7

linux系统安装教程
linux系统安装教程

linux系统是一种可以免费使用,自由传播,多用户、多任务、多线程、多CPU的操作系统。本专题提供linux系统安装教程相关的文章,大家可以免费体验。

2023.07.06

5542

7

linux查看文件夹大小
linux查看文件夹大小

Linux是一种自由和开放源码的类Unix操作系统,存在着许多不同的Linux版本,但它们都使用了Linux内核。Linux可安装在各种计算机硬件设备中,比如手机、平板电脑、路由器、视频游戏控制台、台式计算机、大型机和超级计算机。linux怎么查看文件夹大小呢?php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.07.20

3273

8

linux查看ip命令
linux查看ip命令

本专题为大家提供linux查看ip命令相关文章内容,感兴趣的朋友可以免费下载体验试试。

2023.07.20

2765

4

linux查看cpu使用率
linux查看cpu使用率

在linux的系统维护中,可能需要经常查看cpu使用率,分析系统整体的运行情况。本专题为大家带来了linux查看cpu使用率的相关文章,感兴趣的朋友千万不要错过了。

2023.07.25

2979

6

linux查找文件命令
linux查找文件命令

Linux是一种自由和开放源码的类Unix操作系统,存在着许多不同的Linux版本,但它们都使用了Linux内核。Linux可安装在各种计算机硬件设备中,比如手机、平板电脑、路由器、视频游戏控制台、台式计算机、大型机和超级计算机。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.07.31

277

3

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.6万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 133.4万人学习