Created
March 20, 2017 16:31
-
-
Save tractatus/ce394370026e20624767818acff1711c to your computer and use it in GitHub Desktop.
tensorflow GPU OSX build
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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