Windows编译deepspeed
参考 知乎配置好torch+cuda,vs后,用管理员启动pycharm,设置图中的环境变量以及命令,运行即可编译完成
命令行同理,设置好图中环境变量,运行
python setup.py bdist_wheel
即可,命令如下:
set DS_BUILD_AIO=0
set DS_BUILD_EVOFORMER_ATTN=0
set DS_BUILD_OPS=0
set DS_BUILD_SPARSE_ATTN=0
python setup.py bdist_wheel