跳转到主要内容

Intel® Extension for DeepSpeed*

项目描述

Intel® Extension for DeepSpeed*

Intel® Extension for DeepSpeed* 是一个扩展,它将Intel GPU (XPU) 支持带到了 DeepSpeed (https://github.com/Microsoft/DeepSpeed)。它包含以下组件:

  1. DeepSpeed 加速器接口实现
  2. DeepSpeed op builder XPU 实现
  3. DeepSpeed op builder 内核代码

当 DeepSpeed 作为 Python 包安装时,DeepSpeed 将自动使用 Intel® Extension for DeepSpeed*。安装后,为 DeepSpeed 加速器接口移植并在 DeepSpeed 上运行的模型可以在 Intel GPU 设备上运行。

安装

  1. 按照 安装指南 安装 Intel® Extension for PyTorch*

  2. 安装 Intel® Extension for DeepSpeed*

pip install intel-extension-for-deepspeed

或者,您可以从源代码构建

source ${DPCPPROOT}/env/vars.sh
python setup.py install

通常,对于 root 账户,DPCPPROOT 是 /opt/intel/oneapi/compiler/latest,对于其他账户是 ${HOME}/intel/oneapi/compiler/latest。

  1. 安装 DeepSpeed
pip install deepspeed

开始使用

参考 示例

安全策略

请向英特尔安全中心报告安全问题或漏洞。

有关英特尔如何解决安全问题的更多信息,请参阅漏洞处理指南

项目详情


下载文件

下载适合您平台的文件。如果您不确定选择哪个,请了解更多关于安装包的信息。

源代码分布

此版本没有可用的源代码分布文件。请参阅生成分布存档的教程

构建分布

intel_extension_for_deepspeed-0.9.4-py3-none-any.whl (120.1 kB 查看哈希值)

上传时间 Python 3

由以下提供支持