跳转到主要内容

TensorFlow-DirectML-Plugin是一个开源的DirectML后端,用于TensorFlow。

项目描述

通过TensorFlow在Windows和Windows Subsystem for Linux (WSL)上加速机器学习训练。

TensorFlow-DirectML-Plugin利用TensorFlow的可插拔设备API,为任何DirectX 12兼容的GPU创建DirectML后端,以实现硬件加速的机器学习工作流程。

安装基本的tensorflow和tensorflow-directml-plugin软件包以开始使用。查看GitHub存储库以从源代码构建。

TensorFlow,TensorFlow标志以及任何相关标志都是Google Inc.的商标。

项目详情


下载文件

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

源代码分发

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

构建分发

tensorflow_directml_plugin-0.4.0.dev230202-cp310-cp310-win_amd64.whl (9.0 MB 查看哈希值)

上传时间 CPython 3.10 Windows x86-64

tensorflow_directml_plugin-0.4.0.dev230202-cp310-cp310-manylinux2010_x86_64.whl (26.8 MB 查看哈希值)

上传时间 CPython 3.10 manylinux: glibc 2.12+ x86-64

tensorflow_directml_plugin-0.4.0.dev230202-cp39-cp39-win_amd64.whl (9.0 MB 查看哈希值)

上传时间 CPython 3.9 Windows x86-64

tensorflow_directml_plugin-0.4.0.dev230202-cp39-cp39-manylinux2010_x86_64.whl (26.8 MB 查看哈希值)

上传时间 CPython 3.9 manylinux: glibc 2.12+ x86-64

tensorflow_directml_plugin-0.4.0.dev230202-cp38-cp38-win_amd64.whl (9.0 MB 查看哈希值)

上传时间 CPython 3.8 Windows x86-64

tensorflow_directml_plugin-0.4.0.dev230202-cp38-cp38-manylinux2010_x86_64.whl (26.8 MB 查看哈希值)

上传时间 CPython 3.8 manylinux: glibc 2.12+ x86-64

tensorflow_directml_plugin-0.4.0.dev230202-cp37-cp37m-win_amd64.whl (9.0 MB 查看哈希值)

上传时间 CPython 3.7m Windows x86-64

tensorflow_directml_plugin-0.4.0.dev230202-cp37-cp37m-manylinux2010_x86_64.whl (26.8 MB 查看哈希值)

上传时间 CPython 3.7m manylinux: glibc 2.12+ x86-64

支持者