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
v2.14-v2.18 v1.1.0 12.1+
v2.18 v1.1.0 12.1+ Fixes compatibility with 2.18+ TF versions

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

tensorflow_metal-1.2.0-cp312-cp312-macosx_12_0_arm64.whl (1.4 MB view details)

Uploaded CPython 3.12macOS 12.0+ ARM64

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

Uploaded CPython 3.11macOS 12.0+ ARM64

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

Uploaded CPython 3.10macOS 12.0+ ARM64

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

Uploaded CPython 3.9macOS 12.0+ ARM64

File details

Details for the file tensorflow_metal-1.2.0-cp312-cp312-macosx_12_0_arm64.whl.

File metadata

File hashes

Hashes for tensorflow_metal-1.2.0-cp312-cp312-macosx_12_0_arm64.whl
Algorithm Hash digest
SHA256 4bece0ecb154b713b9f5ad4aec676a366d312822161e3cf0e1dea737c64cec04
MD5 d2cba32ecd1655aeab746111b2c4e1ba
BLAKE2b-256 dcbf988b619322d5617a928e7f31cbb1ed8dd7f375f69dfa73dab26409a00382

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for tensorflow_metal-1.2.0-cp311-cp311-macosx_12_0_arm64.whl
Algorithm Hash digest
SHA256 5fa7cee627031c14f45bd97ff0ef422cd6c3866199ff99cf29b94db6674ceb42
MD5 b019b97d84fbca912d17cf63127a5d19
BLAKE2b-256 e00991b253511cd59b9964672567f36b412daf3c70f75fcb5e84468fafa939ac

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for tensorflow_metal-1.2.0-cp310-cp310-macosx_12_0_arm64.whl
Algorithm Hash digest
SHA256 bc735e36c97874e41f77ec2e7421ff745d2ec36ee641141c8091e4cc2dbcc819
MD5 daf2aab9533b787da5f54007d9163e8c
BLAKE2b-256 39da463240cc8ff13a57119db62a676e2edca86fb93905a13527872dadf1926e

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for tensorflow_metal-1.2.0-cp39-cp39-macosx_12_0_arm64.whl
Algorithm Hash digest
SHA256 4c52a9f7768b10ac226be758e953973734707161caaee328bd2da0cdf93c4643
MD5 48a503834d19d59a9c6bbe288362fdb8
BLAKE2b-256 74c365c82c95f88e4cd949a2ac1ba8a246a438db0a75746694b2c7a8c39524e7

See more details on using hashes here.

Supported by

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