Skip to main content

TensorFlow acceleration for Mac GPUs.

Project description

Accelerate training of machine learning models with TensorFlow right on your Mac. Install the base tensorflow and the tensorflow-metal PluggableDevice to accelerate training with Metal on Mac GPUs.

You can learn more about TensorFlow PluggableDevices here.

tensorflow tensorflow-metal MacOs features
v2.5 v0.1.2 12.0+ Pluggable device
v2.6 v0.2.0 12.0+ Variable seq. length RNN
v2.7 v0.3.0 12.0+ Custom op support
v2.8 v0.4.0 12.0+ RNN perf. improvements
v2.9 v0.5.0 12.1+ Distributed training
v2.10 v0.6.0 12.1+
v2.11 v0.7.0 12.1+
v2.12 v0.8.0 12.1+
v2.13 v1.0.0 12.1+ FP16 and BF16 support

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distributions

If you're not sure about the file name format, learn more about wheel file names.

tensorflow_metal-1.0.1-cp311-cp311-macosx_12_0_arm64.whl (1.4 MB view details)

Uploaded CPython 3.11macOS 12.0+ ARM64

tensorflow_metal-1.0.1-cp310-cp310-macosx_12_0_arm64.whl (1.4 MB view details)

Uploaded CPython 3.10macOS 12.0+ ARM64

tensorflow_metal-1.0.1-cp39-cp39-macosx_12_0_arm64.whl (1.4 MB view details)

Uploaded CPython 3.9macOS 12.0+ ARM64

tensorflow_metal-1.0.1-cp38-cp38-macosx_12_0_arm64.whl (1.4 MB view details)

Uploaded CPython 3.8macOS 12.0+ ARM64

File details

Details for the file tensorflow_metal-1.0.1-cp311-cp311-macosx_12_0_arm64.whl.

File metadata

File hashes

Hashes for tensorflow_metal-1.0.1-cp311-cp311-macosx_12_0_arm64.whl
Algorithm Hash digest
SHA256 bfc7b36a4bed76b00455636cb7c802e98450d6cdc915e5c71405ce492a3e6b3a
MD5 2749a66cb5ed63a42dcace0160ec0d80
BLAKE2b-256 e8a216b46cd175d5e878645ef6d53ce1cd81cb4aa001f2b69ace6c9f72e30dab

See more details on using hashes here.

File details

Details for the file tensorflow_metal-1.0.1-cp310-cp310-macosx_12_0_arm64.whl.

File metadata

File hashes

Hashes for tensorflow_metal-1.0.1-cp310-cp310-macosx_12_0_arm64.whl
Algorithm Hash digest
SHA256 6cd2266b5201b249de2916ab643e72be34440d26716fa62daaeea6442814a9b7
MD5 eca3830489692ed5fcb44a4b508a0bb8
BLAKE2b-256 f33d0796dda099a84e166aacb493f8a161c8816175e514e79012b940364787d4

See more details on using hashes here.

File details

Details for the file tensorflow_metal-1.0.1-cp39-cp39-macosx_12_0_arm64.whl.

File metadata

File hashes

Hashes for tensorflow_metal-1.0.1-cp39-cp39-macosx_12_0_arm64.whl
Algorithm Hash digest
SHA256 236b3585824088a56cda41c79b54bf1ae717bf32c0d7907d5119aef06f81df02
MD5 9ccc742921e71f817811a03e044de1c8
BLAKE2b-256 5de3afd480ccd85304e740918ce37e4120233dd507f64cbb34cae945ddfc9c74

See more details on using hashes here.

File details

Details for the file tensorflow_metal-1.0.1-cp38-cp38-macosx_12_0_arm64.whl.

File metadata

File hashes

Hashes for tensorflow_metal-1.0.1-cp38-cp38-macosx_12_0_arm64.whl
Algorithm Hash digest
SHA256 0c8ae71bec85625b3655ca65810ed26d5f17bd23eb4c2c94e4e49c7e7fa2e5f7
MD5 91ddb04e9a780e6e7426b31eb8154dca
BLAKE2b-256 7e4a17ce1eb4ef0e7ebca58a409f4536fbaffd27f3e88d5a2e4d13afac2a622c

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page