创建DeepSeek-VL2虚拟环境时遇ModuleNotFoundError: No module named 'torch'
解决DeepSeek-VL2虚拟环境依赖安装失败(xformers找不到torch)的问题
步骤1:创建并激活纯净虚拟环境
- 打开VSCode终端,切换到DeepSeek-VL2仓库根目录
- 执行命令创建虚拟环境:
python -m venv venv - 激活虚拟环境:
- Windows:
venv\Scripts\activate - macOS/Linux:
source venv/bin/activate
- Windows:
- 先升级pip到最新版:
pip install --upgrade pip
步骤2:安装适配版本的PyTorch
xformers编译依赖特定版本的PyTorch,必须先确保torch安装成功且版本兼容,推荐安装DeepSeek-VL2适配的PyTorch 2.0.1版本:
- CUDA 11.8用户:
pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118 - CPU用户:
pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cpu - 安装完成后执行
pip list确认torch已出现在依赖列表中
步骤3:调整requirements.txt中的xformers版本
打开仓库根目录的requirements.txt,找到xformers相关行,将其修改为兼容PyTorch 2.0.1的版本:
xformers==0.0.22
(如果原文件中是git链接形式,替换为上述指定版本即可)
步骤4:安装所有依赖
保持虚拟环境激活状态,执行命令安装剩余依赖:
pip install -r requirements.txt
若仍出现xformers编译错误,可单独强制安装指定版本:
pip install xformers==0.0.22 --force-reinstall
额外注意事项
- 确保使用Python 3.8~3.10版本,避免3.11及以上版本导致的兼容性问题
- Windows用户需提前安装Visual Studio 2022构建工具(勾选"C桌面开发"组件),否则无法编译xformers的C扩展
内容的提问来源于stack exchange,提问作者user26598303
相关产品推荐
相关产品推荐

