当前位置: 首页 > article >正文

Windows安装DeepSpeed

文章目录

  • 问题描述
  • 解决方案
  • 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

http://www.kler.cn/a/232032.html

相关文章:

  • 处理namespace问题:Namespace not specified for AGP 8.0.0
  • 跟我学C++中级篇——RAII
  • ADS学习笔记 5. 微带天线设计
  • Python常用魔术方法 (学习笔记)
  • MySQL Online DDL
  • 信息收集—JS框架识别泄露提取API接口泄露FUZZ爬虫插件项目
  • 线阵相机系列-- 1. 什么是线阵相机
  • 【leetcode】深搜、暴搜、回溯、剪枝(C++)1
  • 什么是jieba?
  • 快速熟悉 MatrixOne 内核前端
  • 深度神经网络中的BNN和DNN:基于存内计算的原理、实现与能量效率
  • 【人工智能】人工智能 – 引领未来科技的潮流
  • DataX源码分析 TaskGroupContainer
  • maven java 如何打纯源码zip包
  • 【UE】游戏运行流程的简单理解
  • 网络套件字(理论知识)
  • MySQL用心总结
  • 常见的几种算法排序(C#)
  • 2024-02-04(hive)
  • 缓存异常与一致性
  • 【GoLang入门教程】Go语言几种标准库介绍(五)
  • Linux的Vi编辑器显示行数
  • Vue中的自定义参数校验
  • cesium mapboxgl+threebox glb 朝向问题
  • 跨境电商新风潮:充分发挥海外云手机的威力
  • LeetCode、62.不同路径的数目(一)【简单,动态规划或递归】