How to enable cuda support for tensor flow on Mac OS X (Tensorflow 0.9 on OS X 10.10) - Install tensorflow on OS X with GPU enabled.md. This release version is built for Tensorflow 2.2.0 for Python 3.7 on Mac OS X 10.13 with CUDA support. Supported CUDA compute capability: 3.5,5.0,5.2,6.1,7.0. CUDA 10.0 and CuDNN 7.4 are required. If you need to build your own wheels, use the corresponding source patches and the latest build tutorial in the master branch.
Accelerate training of machine learning models with TensorFlow right on your Mac. Install TensorFlow and the tensorflow-metal PluggableDevice to accelerate training with Metal on Mac GPUs.
Create virtual environment (recommended):
NOTE: python version 3.8 required
Download and install Conda env:
Install the TensorFlow dependencies:
NOTE: python versions 3.8 and 3.9 supported
NOTE: If using conda environment built against pre-macOS 11 SDK use:
SYSTEM_VERSION_COMPAT=0 python -m pip install tensorflow-macos
otherwise you will get errors like : “not a supported wheel on this platform”
To ask questions and share feedback about the tensorflow-metal plugin, visit the Apple Developer Forum.
TensorFlow is tested and supported on the following 64-bit systems:
No install necessary—run the TensorFlow tutorials directly in the browser with Colaboratory, a Google research project created to help disseminate machine learning education and research. It's a Jupyter notebook environment that requires no setup to use and runs entirely in the cloud. Read the blog post.