Windows11下conda虚拟环境安装deepspeed失败求助

核心问题分析 #

报错源于Deepspeed的async_io模块依赖Linux专属的libaio库，Windows系统无法提供该依赖；同时triton库官方仅支持Linux，Windows环境下无法正常安装。

解决方案步骤 #

1. 禁用async_io模块安装 #

根据报错提示，通过设置环境变量禁用async_io的编译：

• CMD命令行：

set DS_BUILD_AIO=0
pip install deepspeed

• PowerShell：

$env:DS_BUILD_AIO=0
pip install deepspeed

2. 同时禁用triton组件（可选） #

若安装时仍遇到triton相关报错，添加禁用triton的环境变量：

• CMD命令行：

set DS_BUILD_AIO=0
set DS_BUILD_TRITON=0
pip install deepspeed

• PowerShell：

$env:DS_BUILD_AIO=0
$env:DS_BUILD_TRITON=0
pip install deepspeed

3. 验证安装结果 #

安装完成后，执行以下命令确认是否成功：

deepspeed --version

正常输出版本信息则说明安装有效。

注意事项 #

Windows环境下Deepspeed的功能存在部分限制，例如部分分布式训练特性无法使用。若需完整功能支持，建议使用WSL2（Windows Subsystem for Linux）搭建Linux环境后安装。

内容的提问来源于stack exchange，提问作者TTTyz