已安装torch但DeepSpeed安装失败求助：提示未检测到torch

问题分析 #

安装DeepSpeed时出现AssertionError: Unable to pre-compile ops without torch installed，但实际已安装PyTorch，核心原因是DeepSpeed的构建脚本无法在临时构建环境中找到当前环境的PyTorch，大概率是环境变量冲突、虚拟环境未激活，或者全局Python环境依赖混乱导致。

解决步骤 #

1. 验证当前PyTorch可用性 #

先确认PyTorch在当前环境中可正常导入：

python -c "import torch; print(torch.__version__)"

如果能正常输出版本号，说明PyTorch已正确安装，继续以下方案。

2. 方案一：禁用预编译Ops快速安装 #

跳过DeepSpeed的Ops预编译过程，绕开依赖检查失败问题：

# Windows系统
set DS_BUILD_OPS=0 && pip install deepspeed

# Linux/macOS系统
export DS_BUILD_OPS=0 && pip install deepspeed

3. 方案二：用虚拟环境隔离依赖 #

若为全局环境依赖混乱导致，创建独立虚拟环境重新配置：

# 创建虚拟环境
python -m venv deepspeed_env

# Windows激活环境
deepspeed_env\Scripts\activate

# Linux/macOS激活环境
# source deepspeed_env/bin/activate

# 重新安装PyTorch（根据CUDA版本调整，以下为CPU默认版）
pip install torch torchvision torchaudio

# 安装DeepSpeed
pip install deepspeed

4. 方案三：安装预编译wheel包 #

直接使用预编译的DeepSpeed包，避免本地编译环节：

pip install deepspeed==0.9.2 --only-binary :all:

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