Distributed package doesn't have NCCL built in:SLURM 作业先确认没有装成 CPU 版

简介: 该问题核心是PyTorch CPU构建(`build CUDA=None`)缺失NCCL,导致`Distributed package doesn't have NCCL built in`错误。集群CUDA模块存在≠PyTorch含NCCL——须在GPU作业节点实测`dist.is_nccl_available()`与`torch.cuda.is_available()`,并换装官方CUDA版PyTorch(如1.12.1+cu116),而非仅改backend或依赖登录节点环境。(239字)

原帖后续 collect_env 明确显示 CUDA used to build PyTorch: None,Conda 构建名为 cpu_py38;这比集群装有 CUDA 11.7 更能解释报错:

RuntimeError: Distributed package doesn't have NCCL built in

作业里的 CPU 构建是已给出的证据

原题 collect_env 的 build CUDA=None 与 cpu_py38 已经指向实际安装不包含 NCCL。集群有 CUDA module 并不能改变 wheel/Conda 包的编译内容。优先在分配 GPU 的作业内核对解释器,避免登录节点与作业节点环境不同。

加载 CUDA module 不会把 CPU torch 变成带 NCCL 的 CUDA 构建。首先在实际 SLURM 作业使用的解释器里检查,而不是只在登录节点测试。

import sys
import torch
import torch.distributed as dist
print("josh050608")
print("python:", sys.executable)
print("torch/build CUDA:", torch.__version__, torch.version.cuda)
print("CUDA available:", torch.cuda.is_available())
print("distributed/NCCL:", dist.is_available(), dist.is_nccl_available())

在分配了 GPU 的作业环境中同时执行 nvidia-smi,核对任务是否拿到设备。原帖这组 CPU 构建需要换成项目兼容的官方 CUDA 构建,不能仅修改 backend 名字。

历史复现可参考 PyTorch 1.12.1 官方页面的 CUDA 11.3/11.6 组合,并核对集群驱动要求与 DeepSpeed/Transformers 约束。例如在独立的兼容 Python 3.8 环境、确认驱动支持 CUDA 11.6 后,可按官方历史组合安装 torch:

python -m pip install torch==1.12.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
python -m pip check

这只给出 torch 的历史构建选择,DeepSpeed 等其余依赖仍按项目约束安装。避免已有 cpuonly、混合 pip/Conda 包继续把解析结果拉回 CPU;当前集群 CUDA module 的数字不等于必须选择完全相同的 wheel 后缀。

若需求仅为 CPU 分布式,可在框架确实支持时选择 Gloo;但原题是 DeepSpeed GPU 微调,不能声称替换成 Gloo 就满足原任务。Windows 官方 NCCL 支持限制也不应套到这道已确认 Linux 的题上。

验证依次为 is_nccl_available=True、单卡分配、两进程小 collective,再启动真正训练。无需为了验证安装立即重跑完整微调。

安装能力与设备可见性分别验证

dist.is_nccl_available 检查当前构建是否带 NCCL;CUDA 设备分配检查作业是否拿到可用 GPU。两者都要通过,再进入 DeepSpeed,而非用 backend='gloo' 掩盖 GPU 微调需求。

下面脚本可放到实际 SLURM 作业的诊断阶段,不初始化分布式训练,也不运行模型:

import os
import sys
import torch
import torch.distributed as dist
print("job/local rank:", os.environ.get("SLURM_JOB_ID"), os.environ.get("LOCAL_RANK"))
print("python/build:", sys.executable, torch.__version__, torch.version.cuda)
print("NCCL built:", dist.is_nccl_available())
print("visible/count:", os.environ.get("CUDA_VISIBLE_DEVICES"), torch.cuda.device_count())
if torch.cuda.is_available():
    test = torch.ones(1, device="cuda:0")
    torch.cuda.synchronize()
    print("allocation:", test.device)

如果换包后仍得到 build CUDA=None,检查作业脚本激活的环境、cpuonly 约束以及 pip/Conda 混装结果。若构建带 CUDA/NCCL 但设备不可用,转查调度分配、驱动和容器映射,不再重复修包名。

torch 修复后还要核对 DeepSpeed 与 Transformers 对应版本,尤其自编译扩展的工具链。预编译运行库与系统 nvcc 不要求永远逐字相等,扩展编译则另按项目要求检查。

最后做两进程最小 collective,再用极小训练批次验证 DeepSpeed 初始化。无需立即启动完整 ProtGPT-2 微调;构建检查通过不等于分布式通信和训练全部通过。

参考:原始求助帖;技术依据:PyTorch 分布式(2.5)、PyTorch 历史安装组合、PyTorch CUDA 语义(2.5)、Transformers Trainer(4.46)。

相关文章
|
10天前
|
Linux 开发工具 iOS开发
Codex保姆级教程(下载+安装+登录+使用,含所有版本,2026最新)
Codex是OpenAI推出的开源AI编程智能体,本地运行,可理解整个代码库,支持自然语言指令改代码、执行命令、跑测试、修Bug,兼容Win/macOS/Linux。(239字)
3995 19
|
11天前
|
JavaScript API iOS开发
DeepSeek Harness下载、安装、配置一篇搞定(2026整理,送常用插件)
DeepSeek Harness(DSH)是DeepSeek于2026年开源的MIT协议AI Agent框架,支持插件化组装模型、工具、沙箱与UI,兼容主流大模型。本文提供免Node.js的桌面端安装包(Win/macOS),含详细图文教程及热门插件推荐。
4135 49
|
11天前
|
iOS开发 Docker MacOS
【2026实测】Docker Desktop下载、安装、汉化、配置教程(附官网安装包)
Docker Desktop 是 Docker 官方推出的桌面容器管理工具,集成引擎、CLI、图形界面与 Kubernetes,支持 Win/macOS/Linux。2026 最新版 4.88.1 强化 ARM 原生运行、镜像加速及资源可视化,附中文汉化与国内镜像配置指南。
1976 7
|
6天前
|
云安全 人工智能 安全
|
6天前
|
存储 弹性计算 人工智能
2026 年阿里云双十一活动入口攻略,新客服务器优惠通道、三大会场入口梳理
综合来看,2026年阿里云双十一活动入口清晰: 阿里云轻量应用服务器、阿里云ECS云服务器、阿里云GPU云服务器、阿里云百炼大模型服务平台 活动会场、综合权益中心、99计划专场三大入口覆盖服务器、AI权益与续费特价;新用户通过控制台首页banner、权益中心、产品页三处路径进入专属专场,老用户通过控制台弹窗和费用中心权益中心进入老客专区,企业用户做企业实名后解锁专属满减与阶梯折扣。
819 0
|
5天前
|
Linux 开发工具 iOS开发
2026最新codex下载安装保姆级使用教程(下载+安装+登录+使用,含所有历史版本)
本文详解OpenAI Codex国内快速上手指南:提供Win/macOS/Linux全平台免安装包、npm/Homebrew等多方式安装、账号/API Key双登录、简体中文切换,覆盖下载、安装、汉化、使用及高频避坑(如微软商店失败、npm超时等),10分钟即可跑通。
1087 2
|
12天前
|
安全 算法 定位技术
《高性价比人生指南》完整原版:普通人用最少的资源,过好这一生的底层逻辑~How To Live Better
📖《高性价比人生指南》是一份基于实证、拒绝焦虑的生存手册,倡导以“人生时薪”为视角,重新计算时间、金钱与精力的成本与收益。它不承诺逆袭,只教人算清账:省下的钱若耗掉时间便亏;买体验比买物品更值;睡眠是最高杠杆的健康投资;关系贵在情绪价值而非表面热络;学习重在改变行为而非自我安慰。核心理念:用低成本、高回报的微小行动,换取长期的选择权——不是成为谁,而是掌控自己人生的主动权。
2144 6
|
云安全 安全 网络安全
|
24天前
|
人工智能 API 开发工具
DeepSeek Harness 神插件:白嫖 13 家大厂模型
`dsh-our-free-model` 是 DeepSeek Harness 神级插件:零配置、零API Key、零账号,聚合13家大厂免费模型(含DeepSeek V4.1 Flash、Kimi K3等),支持本地网关、用量看板与自动降级,真正实现“白嫖即用”。
4337 58