1 配置基础环境
anaconda创建虚拟环境并安装pytorch+cuda,随后配置环境变量。
2 安装Visual Studio 2022
工作负荷选择“使用C++的桌面开发”。
3 使用Git克隆DeepSpeed文件
git clone https://github.com/microsoft/DeepSpeed.git
4 运行Anaconda PowerShell
以管理员身份运行,先激活创建好的虚拟环境,随后cd进入DeepSpeed文件夹下。
cd DeepSpeed
然后运行build_win.bat编译DeepSpeed。
5 安装DeepSpeed
进入dist文件夹,运行pip install命令。
pip install deepspeed-0.13.5+07a1ede4-py3-none-any.whl
6 报错解决
6.1 Unable to pre-compile
使用如下命令。
Set-Item Env:\DS_BUILD_OPS 0