Off-the-shelf python package of tensorflow with CUDA support for Mac OS.
The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:
Unfortunately, the Tensorflow team stops releasing binary package for Mac OS with CUDA support since Tensorflow 1.2. This project provides off-the-shelf binary packages.
Both Python 2.7 and 3.7 are supported now!
很不幸，Tensorflow团队自从1.2版本开始停止了发布 Mac OS CUDA版。本项目提供 Mac OS 上编译好、可直接安装的Tensorflow CUDA版本。
本项目同时支持Python 2.7 和 3.7 了！
You can find releases in the releases page.
Except for making patches for release verions of TF, I fork TF sources at https://github.com/TomHeaven/tensorflow and keep fixing build issues of TF on macOS with CUDA. The corresponding PR is at: https://github.com/tensorflow/tensorflow/pull/39297. You can use the PR to make your own builds.
First, ensure your CUDA driver and cudnn is installed properly, and copy dependencies in folder
sudo mkdir /usr/local sudo mkdir /usr/local/lib sudo cp usr_local_lib/* /usr/local/lib/
Second, uninstall the previous tensorflow installtion by
pip uninstall tensorflow pip uninstall tensorflow-gpu # for early version with offical support
At last, download binary packages from Releases page and install
pip install tensorflow*.whl
Install Python 3.7 from Homebrew first, and then simply follow the guide for Python 2.7 and replace
首先从Homebrew安装Python 3.7，然后按照Python 2.7的安装步骤执行，注意将
If you want to build your own wheel packages, refer to the following tutorials:
If you need Pytorch builds for osx, go to this page: https://github.com/TomHeaven/pytorch-osx-build
If you need MxNet builds for osx, go to this page: https://github.com/TomHeaven/mxnetosxbuild