Skip to content

Instantly share code, notes, and snippets.

@dongyuwei
Created October 16, 2016 13:30
Show Gist options
  • Save dongyuwei/259bb2cb7e79de1a9f03002cb429fc1d to your computer and use it in GitHub Desktop.
Save dongyuwei/259bb2cb7e79de1a9f03002cb429fc1d to your computer and use it in GitHub Desktop.
source cuda.sh before load tensorflow in python
######### for tensorflow and cuda ##############################################################
# sudo cp /usr/local/cuda/lib/libcuda.dylib /usr/local/cuda/lib/libcuda.1.dylib
export CUDA_HOME=/usr/local/cuda
export DYLD_LIBRARY_PATH="$DYLD_LIBRARY_PATH:$CUDA_HOME/lib"
export PATH="$CUDA_HOME/bin:$PATH"
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib:/usr/local/cuda/extras/CUPTI/lib/"
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment