Skip to content

Instantly share code, notes, and snippets.

@tractatus
Created March 20, 2017 16:31
Show Gist options
  • Save tractatus/ce394370026e20624767818acff1711c to your computer and use it in GitHub Desktop.
Save tractatus/ce394370026e20624767818acff1711c to your computer and use it in GitHub Desktop.
tensorflow GPU OSX build
brew install python3
brew install coreutils
brew tap caskroom/cask
brew cask install cuda
nano ~/.bash_profile
#paste in the following
export CUDA_HOME=/usr/local/cuda
export DYLD_LIBRARY_PATH=${CUDA_HOME}/lib:${CUDA_HOME}/extras/CUPTI/lib:/Developer/NVIDIA/CUDA-8.0/lib
export LD_LIBRARY_PATH=$DYLD_LIBRARY_PATH
export PATH=$DYLD_LIBRARY_PATH:$PATH
# install the CUDA Deep Neural Network (cuDNN v5) library
# First register for a member:
#https://developer.nvidia.com/cudnn
#Accelerated Computing Developer Program account.
#Once you have it downloaded locally (https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v5.1/prod_20161129/8.0/cudnn-8.0-osx-x64-v5.1-tgz) CUDA Deep Neural Network (cuDNN v5), you can unzip and move the header and libraries to your local CUDA Toolkit folder:
cd ~/Downloads/cuda
sudo mv include/cudnn.h /Developer/NVIDIA/CUDA-8.0/include/
sudo mv lib/libcudnn* /Developer/NVIDIA/CUDA-8.0/lib
sudo ln -s /Developer/NVIDIA/CUDA-8.0/lib/libcudnn* /usr/local/cuda/lib/
sudo python3 -m pip install --upgrade https://storage.googleapis.com/tensorflow/mac/grpu/tensorflow-0.11.0-py3-none-any.whl
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment