Tensorflow supporting CUDA for MacOS

CUDAをサポートしたTensorflowは自分でビルドする必要がある。ビルドしたものを以下にアップロードしています。

https://storage.googleapis.com/74thopen/tensorflow_osx/index.html

v1.5.0

pip3 install https://storage.googleapis.com/74thopen/tensorflow_osx/tensorflow-1.5.0-cp36-cp36m-macosx_10_13_x86_64.whl
  • 2018/02/03
  • Tensorflow: v1.5.0
  • OSX: 10.13.3(High Sierra)
  • XCode: 8.2
  • CUDA Driver: 9.1(387.128)
  • Cuda compute capabilities: 6.1 (list)
  • cuDNN: 7
  • GPU Driver Version: 387.10.10.10.25.156
  • OpenMP (LLVM v5.0.0)
  • Python: 3.6.4

eGPU NVIDIA Driverのインストール

https://egpu.io/forums/mac-setup/wip-nvidia-egpu-support-for-high-sierra/

OpenMP のインストール

wget http://releases.llvm.org/5.0.0/openmp-5.0.0.src.tar.xz
xz -dc openmp-5.0.0.src.tar.xz | tar xfv -
cd openmp-5.0.0.src
mkdir build
cd build
cmake ..
make install

v1.4.0

pip3 install https://storage.googleapis.com/74thopen/tensorflow_osx/tensorflow-1.4.1-cp36-cp36m-macosx_10_13_x86_64.whl
  • 2017/11/25
  • OSX: 10.13(High Sierra)
  • XCode: 8.2.1
  • CUDA Driver: 9.0.22
  • Tensorflow: v1.4.0