跳转到主要内容

TensorFlow为Mac GPU加速。

项目描述

在您的Mac上使用TensorFlow加速机器学习模型的训练。安装基本的tensorflow和tensorflow-metal PluggableDevice以在Mac GPU上加速训练。

您可以在这里了解更多关于TensorFlow PluggableDevices的信息。

tensorflow tensorflow-metal MacOs 功能
v2.5 v0.1.2 12.0+ 可插拔设备
v2.6 v0.2.0 12.0+ 可变序列长度RNN
v2.7 v0.3.0 12.0+ 自定义操作支持
v2.8 v0.4.0 12.0+ RNN性能改进
v2.9 v0.5.0 12.1+ 分布式训练
v2.10 v0.6.0 12.1+
版本 2.11 版本 0.7.0 12.1+
版本 2.12 版本 0.8.0 12.1+
版本 2.13 版本 1.0.0 12.1+ 支持 FP16 和 BF16
版本 2.14 版本 1.1.0 12.1+

项目详情


下载文件

下载适用于您平台的应用程序。如果您不确定选择哪个,请了解有关 安装包 的更多信息。

源代码分发

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

构建分发

tensorflow_metal-1.1.0-cp311-cp311-macosx_12_0_arm64.whl (1.4 MB 查看哈希值)

上传时间 CPython 3.11 macOS 12.0+ ARM64

tensorflow_metal-1.1.0-cp310-cp310-macosx_12_0_arm64.whl (1.4 MB 查看哈希值)

上传时间 CPython 3.10 macOS 12.0+ ARM64

tensorflow_metal-1.1.0-cp39-cp39-macosx_12_0_arm64.whl (1.4 MB 查看哈希值)

上传时间 CPython 3.9 macOS 12.0+ ARM64

支持者

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