1. 技术文章
  2. tensorflow Mac OS gpu support

tensorflow Mac OS gpu support

由 故事扮演 提交于 2019-11-29 19:37:54

I wrote a little tutorial on compiling TensorFlow 1.2 with GPU support on macOS. I think it's customary to copy relevant parts to SO, so here it goes:

  1. If you haven’t used a TensorFlow-GPU set-up before, I suggest first setting everything up with TensorFlow 1.0 or 1.1, where you can still do pip install tensorflow-gpu. Once you get that working, the CUDA set-up would also work if you’re compiling TensorFlow. If you have an external GPU, YellowPillow's answer (or mine) might help you get things set up.
  2. Follow the official tutorial “Installing TensorFlow from Sources”, but obviously substitute git checkout r1.0 with git checkout r1.2. When doing ./configure, pay attention to the Python library path: it sometimes suggests an incorrect one. I chose the default options in most cases, except for: Python library path, CUDA support and compute capacity. Don’t use Clang as the CUDA compiler: this will lead you to an error “Inconsistent crosstool configuration; no toolchain corresponding to 'local_darwin' found for cpu 'darwin'.”. Using /usr/bin/gcc as your compiler will actually use Clang that comes with macOS / XCode. Below is my full configuration.
  3. TensorFlow 1.2 expects a C library called OpenMP, which is not available in the current Apple Clang. It should speed up multithreaded TensorFlow on multi-CPU machines, but it will also compile without it. We could try to build TensorFlow with gcc 4 (which I didn’t manage), or simply remove the line that includes OpenMP from the build file. In my case I commented out line 98 of tensorflow/third_party/gpus/cuda/BUILD.tpl, which contained linkopts = [“-lgomp”] (but the location of the line might obviously change). Some people had issues with zmuldefs, but I assume that was with earlier versions; thanks to udnaan for pointing out that it’s OK to comment out these lines.
  4. I had some problems building with the latest bazel 0.5.3, so I reverted to using 0.4.5 that I already had installed. But some discussion in a github issue mentioned bazel 0.5.2 also didn’t have the problem.
  5. Now build with bazel and finish the installation as instructed by the official install guide. On my 3.2 GHz iMac this took about 37 minutes.

Using python library path: /Users/m/code/3rd/conda/envs/p3gpu/lib/python3.6/site-packages

Do you wish to build TensorFlow with MKL support? [y/N] N

No MKL support will be enabled for TensorFlow

Please specify optimization flags to use during compilation when bazel option "--config=opt" is specified [Default is -march=native]:

Do you wish to build TensorFlow with Google Cloud Platform support? [y/N]

No Google Cloud Platform support will be enabled for TensorFlow

Do you wish to build TensorFlow with Hadoop File System support? [y/N]

No Hadoop File System support will be enabled for TensorFlow

Do you wish to build TensorFlow with the XLA just-in-time compiler (experimental)? [y/N]

No XLA support will be enabled for TensorFlow

Do you wish to build TensorFlow with VERBS support? [y/N]

No VERBS support will be enabled for TensorFlow

Do you wish to build TensorFlow with OpenCL support? [y/N]

No OpenCL support will be enabled for TensorFlow

Do you wish to build TensorFlow with CUDA support? [y/N] y

CUDA support will be enabled for TensorFlow

Do you want to use clang as CUDA compiler? [y/N]

nvcc will be used as CUDA compiler

Please specify the CUDA SDK version you want to use, e.g. 7.0. [Leave empty to use system default]:

Please specify the location where CUDA toolkit is installed. Refer to README.md for more details. [Default is /usr/local/cuda]:

Please specify which gcc should be used by nvcc as the host compiler. [Default is /usr/bin/gcc]:

Please specify the cuDNN version you want to use. [Leave empty to use system default]:

Please specify the location where cuDNN library is installed. Refer to README.md for more details. [Default is /usr/local/cuda]:

Please specify a list of comma-separated Cuda compute capabilities you want to build with.

You can find the compute capability of your device at: https://developer.nvidia.com/cuda-gpus.

Please note that each additional compute capability significantly increases your build time and binary size.

[Default is: "3.5,5.2"]: 6.1

INFO: Starting clean (this may take a while). Consider using --async if the clean takes more than several minutes.

Configuration finished

Assuming that you have already setup your eGPU box and attached the TB3 cable from the eGPU to your TB3 port:

1. Download the automate-eGPU script and run it

curl -o ~/Desktop/automate-eGPU.sh
https://raw.githubusercontent.com/goalque/automate-eGPU/master/automate-eGPU.sh
&& chmod +x ~/Desktop/automate-eGPU.sh && cd ~/Desktop && sudo
./automate-eGPU.sh

You might get an error saying:

"Boot into recovery partition and type: csrutil disable"

All you need to do now is to restart your computer and when it's restarting hold down cmd + R to enable the recovery mode. Then locate the Terminal while in recovery mode and type in:

csrutil disable

Then restart your computer and re-run the automate-eGPU.sh script

2: Download and installing CUDA

Run the cuda_8.0.61_mac.dmg file and follow through the installation phase. Then afterwards you will need to set the paths.

Go to your Terminal and type:

vim ~/.bash_profile

Or whether you have stored your environmental variables and then add these three lines:

export CUDA_HOME=/usr/local/cuda
export DYLD_LIBRARY_PATH="$CUDA_HOME/lib:$CUDA_HOME:$CUDA_HOME/extras/CUPTI/lib"
export LD_LIBRARY_PATH=$DYLD_LIBRARY_PATH

3. Downloading and installing cuDNN

To download cuDNN is a bit more troublesome you have to sign up to be a developer for Nvidia and then afterwards you can download it. Make sure to download cuDNN v5.1 Library for OSX as it's the one that Tensorflow v1.1 expects Note that we can't use Tensorflow v1.2 as there is no GPU support for Macs :((

[![enter image description here][1]][1]

Now you will download a zip file called cudnn-8.0-osx-x64-v5.1.tgz, unzip and, which will create a file called cuda and cd to it using terminal. Assuming that the folder is in Downloads

Open terminal and type:

cd ~/Downloads/cuda

Now we need to copy cuDNN files to where CUDA is stored so:

sudo cp include/* /usr/local/cuda/include/
sudo cp lib/* /usr/local/cuda/lib/

4. Now install Tensorflow-GPU v1.1 in your conda/virtualenv

For me since I use conda I created a new environment using Terminal:

conda create -n egpu python=3
source activate egpu
pip install tensorflow-gpu # should install version 1.1

5. Verify that it works

First you have to restart your computer then:

In terminal type python and enter:

import tensorflow as tf
with tf.device('/gpu:0'):
    a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
    b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
    c = tf.matmul(a, b)

with tf.Session() as sess:
    print (sess.run(c))

If you have a GPU this should run with no problem, if it does then you should get a stack trace (just a bunch of error messages) and it should include

Cannot assign a device to node 'MatMul': Could not satisfy explicit device specification '/device:GPU:0' because no devices matching that specification are registered in this process

If not then you're done congratz! I just got mine set up today and it's working perfectly :)

I could finally make it work with the following setup

Hardware

Software versions

  • macOS Sierra Version 10.12.6
  • GPU Driver Version: 10.18.5 (378.05.05.25f01)
  • CUDA Driver Version: 8.0.61
  • cuDNN v5.1 (Jan 20, 2017), for CUDA 8.0: Need to register and download
  • tensorflow-gpu 1.0.0
  • Keras 2.0.8

I wrote a gist with the procedure:

https://gist.github.com/jganzabal/8e59e3b0f59642dd0b5f2e4de03c7687

Here is my solution to install an e-gpu on a mac. Tensorflow doesn't support tensorflow-gpu anymore, so there are definitely better approaches to get it working:

My configuration:

  • IMac 27' late 2012
  • Aktio Node
  • GTX 1080 ti
  • 3 Screens: One of them connected to the GTX 1080 and the others directly plugged on the mac.

Advantages of windows bootcamp installation:

  • You can use pip to install tensorflow-gpu.
  • Good GPU 1080 ti support (Downloadable display driver)

Howto:

  • Install windows 10 with bootcamp. Do not connect the Akito node for the moment.
  • Download and install the display driver for your gpu from NVIDIA download page
  • Install Visual Studio
    • If you want to use CUDA 9.x you can install Visual Studio 2017
    • Otherwise install Visual Studio 2015
  • Install CUDA and CuDNN
    • Note that the tensorflow-gpu version must match with your Cuda and your CudNN version. See available tensorflow releases here.
    • After the CUDA installation you can move the unpacked CuDNN files to the CUDA folder at: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0. Move the lib files to the lib folder, the bin files to the bin folder and the include files to the include folder.
  • Install Python 3.5+
    • You need a 64-bit version to install tensorflow-gpu with pip
    • Python 2.7 won't work.
  • Install tensorflow with pip:

Command:

pip install tensorflow-gpu==1.5.0rc0

Check your installation

The display driver has been installed correctly when you can plug a screen to the GTX 1080 ti card.

Call C:\Program Files\NVIDIA Corporation\NVSMI\nvidia-smi.exe to check if your video card is available for CUDA.

Execute the following tensorflow command to see available devices:

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

Troubleshooting and hints:

  • Windows wants to update your GTX 1080 driver. Never allow that because you won't be able to startup your computer again! A black screen with moving dots will appear before you can login to windows. Game over! Only use the display driver from NVIDIA download page.
  • If you cannot start windows on OSX anymore, press the alt key at startup to reinstall windows.

Ubuntu solution:

I couldn't find a working solution but here are some approaches:

It seems that my GTX 680 (iMac) and my GTX 1080 ti won't work together. Ubuntu could not be started anymore after installing the display driver via apt-get: Ubuntu not starting anmore. Try to download the official display driver from NVIDIA download page.

OSX Solution: Tensorflow GPU is only supported up to tensorflow 1.1. I tried to install a newer version but couldn't build tensorflow-gpu with cuda support. Here are some approaches:

  • Install OSX Sierra to use the e-gpu script. High Sierra won't work (Jan, 13 2018). Downgrade to sierra by deleting all your partitions. Then press Command + R at startup to load the internet recovery. Don't forget to backup your data first.
  • Install e-gpu script.
  • If tensorflow-gpu 1.1 is enough for you, you can just install via pip, otherwise you need to build your pip with bazel.

Conclusion: The windows installation is easier than OSX or Ubuntu installation because display drivers work properly and tensorflow and must not be build on your own. Always check the software version you use. The must match exactly.

I hope this will help you!

标签
macos
tensorflow
tensorflow-gpu
易学教程内所有资源均来自网络或用户发布的内容，如有违反法律规定的内容欢迎反馈！
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!