Skip to content

Instantly share code, notes, and snippets.

@chexov
Forked from smitshilu/Tensorflow_Build_GPU.md
Created December 11, 2017 23:54
Show Gist options
  • Select an option

  • Save chexov/cefb960264d1c7b17a8ebe9e7da98f0b to your computer and use it in GitHub Desktop.

Select an option

Save chexov/cefb960264d1c7b17a8ebe9e7da98f0b to your computer and use it in GitHub Desktop.
Tensorflow 1.4 Mac OS High Sierra 10.13 GPU Support

Tensorflow

System information

  • OS - High Sierra 10.13
  • Tensorflow - 1.4
  • Xcode command line tools - 7.2 (Download from here: Xcode - Support - Apple Developer & Switch to different clang version: sudo xcode-select --switch/Library/Developer/CommandLineTools & check version: clang -v)
  • Cmake - 3.7
  • Bazel - 0.7.0
  • CUDA - 9
  • cuDNN - 7

Requirements

  • sudo pip install six numpy wheel
  • brew install coreutils

Step-by-step guide

Remove all align(sizeof(T)) from following files:

  • tensorflow/core/kernels/depthwise_conv_op_gpu.cu.cc

  • tensorflow/core/kernels/split_lib_gpu.cu.cc

  • tensorflow/core/kernels/concat_lib_gpu.impl.cu.cc

    For example, extern shared __align(sizeof(T))__ unsigned char smem[]; => extern shared unsigned char smem[];

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment