admin 管理员组

文章数量: 1184232

文章目录

  • 问题描述
  • 解决方案
  • AssertionError: Unable to pre-compile async_io
  • 参考文献

问题描述

DeepSpeed 是一款微软推出的深度学习优化库，它使分布式训练和推理变得简单高效。




解决方案

需要提前安装：

  • PyTorch，版本最好 >= 1.9
  • CUDA 或 ROCm 编译器，如 nvcc 或 hipcc
  • 安装 Visual Studio，如 2019

自行编译

python setup.py bdist_wheel

进入 dist 文件夹安装该 wheel

pip install xxx




AssertionError: Unable to pre-compile async_io

修改 setup.py

BUILD_OP_PLATFORM = 1 if sys.platform == "win32" else 0

为

BUILD_OP_PLATFORM = 0




参考文献

  1. DeepSpeed GitHub
  2. Win10安装DeepSpeed

本文标签： Windows deepspeed

版权声明：本文标题：Windows安装DeepSpeed 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://roclinux.cn/b/1765934027a3425598.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。