跳转到主要内容

PyTorch的DirectML后端,用于硬件加速。

项目描述

PyTorch with DirectML

DirectML为PyTorch提供的加速功能目前处于公开预览阶段。PyTorch with DirectML允许在广泛的DirectX 12兼容硬件上训练和推理复杂的机器学习模型。

DirectML是一个高性能、硬件加速的DirectX 12库,用于机器学习。DirectML为支持硬件和驱动程序上常见的机器学习任务提供GPU加速,包括来自AMD、Intel、NVIDIA和Qualcomm等所有DirectX 12兼容GPU的供应商。

有关DirectML的更多信息,请参阅 DirectML简介

PyTorch在DirectML的支持下,既适用于Windows 10的最新版本,也适用于Windows Subsystem for Linux,并且作为PyPI软件包可供下载。有关入门信息,请参阅GPU加速ML训练(docs.microsoft.com)

示例

请参考带有DirectML的PyTorch示例存储库

反馈

我们期待您的反馈!

贡献

本项目欢迎贡献和建议。大多数贡献需要您同意贡献者许可协议(CLA),声明您有权并且确实授予我们使用您贡献的权利。有关详细信息,请访问https://cla.microsoft.com

在您提交拉取请求时,CLA机器人将自动确定您是否需要提供CLA,并相应地装饰PR(例如,标签、注释)。只需遵循机器人提供的说明即可。您只需要在整个使用我们的CLA的存储库中做一次。

本项目已采用Microsoft开源行为准则。有关更多信息,请参阅行为准则FAQ或通过opencode@microsoft.com联系我们有任何额外的问题或评论。

工作流程

开发者将每天在发布分支上工作,并在需要时将RI推送到主分支。

为何采用这种模式的动机。

  1. 避免两次将相同的更改推送到发布和主分支。
  2. 避免被来自pytorch主分支的破坏性更改阻塞。

剪枝分支的例子

  1. 为pytorch-1.13创建新分支(例如,[release/1.13+2])将从最后一个发布的分支release/1.13+剪出,并且开发人员将每天停留在新分支上,直到该版本发布。

  2. 对于新的pytorch版本1.13.1,应在所有发布工作负载都RI之后从主分支中剪出release/1.13.1。

workflow

外部链接

PyTorch主页

项目详情


下载文件

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

源分发

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

构建分发

experimental_torch_directml-0.1.13.dev23122022132839-cp39-cp39-win_amd64.whl (7.4 MB 查看哈希值)

上传时间 CPython 3.9 Windows x86-64

由以下机构支持

AWS AWS 云计算和安全赞助商 Datadog Datadog 监控 Fastly Fastly CDN Google Google 下载分析 Microsoft Microsoft PSF 赞助商 Pingdom Pingdom 监控 Sentry Sentry 错误日志 StatusPage StatusPage 状态页面