• Stars
    star
    142
  • Rank 257,199 (Top 6 %)
  • Language
  • Created almost 7 years ago
  • Updated almost 4 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Off-the-shelf python package of tensorflow with CUDA support for Mac OS.

Tensorflow OSX Build

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 了!

Releases

You can find releases in the releases page.

你可以在Releases页面找到以前发布的版本。

My Fork of Tensorflow

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: tensorflow/tensorflow#39297. You can use the PR to make your own builds.

Installation for Python 2.7

First, ensure your CUDA driver and cudnn is installed properly, and copy dependencies in folder usr_local_lib to path /usr/local/lib.

首先,确保CUDA驱动和cudnn正确安装,并且将文件夹usr_local_lib中的依赖项复制到路径/usr/local/lib

sudo mkdir /usr/local
sudo mkdir /usr/local/lib
sudo cp usr_local_lib/* /usr/local/lib/

Second, uninstall the previous tensorflow installtion by

其次,卸载之前版本的Tensorflow:

pip uninstall tensorflow
pip uninstall tensorflow-gpu # for early version with offical support

At last, download binary packages from Releases page and install

最后,从Releases页面下载并安装:


pip install tensorflow*.whl

Installation for Python 3.7

Install Python 3.7 from Homebrew first, and then simply follow the guide for Python 2.7 and replace pip command with pip3 and python with python3.

首先从Homebrew安装Python 3.7,然后按照Python 2.7的安装步骤执行,注意将pip替换为pip3,并用python3启动python

Enjoy!

开始使用新版Tensorflow吧!

Build Tutorial

If you want to build your own wheel packages, refer to the following tutorials:

  • v1.10
  • v2.0.0 This tutorial also works for v1.15.0, just use source patch v1.15.0 instead of v2.0.0.

Related Links

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/mxnet_osx_build

如果你需要Pytorch包,请看这个页面:https://github.com/TomHeaven/pytorch-osx-build

如果你需要MxNet包,请看这个页面:https://github.com/TomHeaven/mxnet_osx_build

More Repositories

1

nudt_thesis

NUDT硕士博士毕业论文latex模板
TeX
161
star
2

pytorch-osx-build

Off-the-shelf python package of pytorch with CUDA support for Mac OS
C
145
star
3

HoRNDIS

Tom's Repo of HoRNDS——the USB tethering driver for Mac OS 11+
C++
101
star
4

nudtproposal

NUDT硕士博士研究生开题报告latex模板
TeX
82
star
5

Joint-Demosaic-and-Denoising-with-ADMM

[2017] Code for ICIP 2017 paper JOINT DEMOSAICING AND DENOISING OF NOISY BAYER IMAGES WITH ADMM
MATLAB
64
star
6

msba_reid

[2020] Official code for paper "MSBA: Multiple Scales, Branches and Attention Network With Bag of Tricks for Person Re-Identification"
Python
23
star
7

AnomalyAnalysisWithOpticalFlow

[2016] A project for video anomaly detection of our ICASSP 2016 paper.
C++
20
star
8

wiznote-release-for-linux

为知笔记Linux版
Shell
15
star
9

Dark-Channel-Haze-Removal-with-CUDA

Dark channel Haze removal algorithm with CUDA acceleration (typically 10x or more speedup using a Nvidia GPU)
Cuda
14
star
10

Pixel-wise-Estimation-of-Signal-Dependent-Image-Noise-using-Deep-Residual-Learning

Code of our paper Pixel-wise Estimation of Signal Dependent Image Noise using Deep Residual Learning
Python
14
star
11

Sparse-Coding-based-Image-Super-Resolution-with-CUDA

Code for ICIVC 2017 paper LASSO Approximation and Application to Image Super-resolution with CUDA Acceleration.
Cuda
7
star
12

nccl-osx

Nvidia's NCCL library migrated to Mac OS X (10.13-10.13.6)
C++
3
star
13

MacKernelSDK

macOS kernel SDK targeting various XNU versions (Support both Intel CPU & Apple Silicon)
C
3
star
14

cvxpy

Python
1
star
15

crypten_cifar10_cuda_distributed

A real distributed runnable demonstration of CrypTen image classification on Cifar10 dataset with CUDA enabled.
Python
1
star
16

netcdf2json

Convert netcdf4 ocean current data to json for Null School Earth project.
Python
1
star