Macbook Pro 外接显卡实现Tensorflow GPU运行之TensorflowGPU配置

Macbook Pro 外接显卡实现Tensorflow GPU运行

1. MacOS 重装
2. 外接显卡（内屏输出）
3. 环境配置（重点）
4. Tensorflow GPU 配置 👆

---

Tensorflow GPU配置

准备工作

• 硬件准备：外接显卡，外接显示器/显卡欺骗器
• 环境/软件准备：相关驱动和环境编译工具一定要确保编译成功哦！
  以笔者为例：

资源	版本号
MacOS	10.13. 6（17G65）
GPU driver	NIVIDIA Web Driver 387.10.10.40.105
CUDA driver	9.2（396.148 ）
cuDNN	cudnn-9.2-osx-x64-v7.2.1.38
Xcode	9.2
Python	3.6
tensorflow	1.8/1.4

Pyhton依赖包：
six，numpy，wheel，Coreutils，llvm，OpenMP，bazel（0.14.0）

安装步骤

1. 确认环境编译通过后，新建虚拟环境，安装python依赖包：
   进入terminal，通过pip或conda，brew依次安装以上包安装
   例如：

$ conda create -n tf_gpu_py3.6
$ source activate tf_gpu_py3.6
$ conda install six 

$ brew install coreutils llvm cliutils/apple/libomp

安装bazel 0.14.0 https://github.com/bazelbuild/bazel/releases

$ curl -O https://github.com/bazelbuild/bazel/releases/download/0.14.0/bazel-0.14.0-installer-darwin-x86_64.sh
$ chmod +x bazel-0.14.0-installer-darwin-x86_64.sh
$ ./bazel-0.14.0-installer-darwin-x86_64.sh
$ bazel version
Build label: 0.14.0

2. 下载安装python3.6

$ conda install python=3.6

3. 下载安装tensorflow-gpu 1.8
   release 1.8 安装：
   直接下载修改好的源码

$ curl -O https://raw.githubusercontent.com/SixQuant/tensorflow-macos-gpu/master/tensorflow-macos-gpu-r1.8-src.tar.gz

配置：

$ which python3
/usr/local/bin/python3

$ ./configure

开始编译：

$ bazel clean --expunge
$ bazel build --config=opt --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" --action_env PATH --action_env DYLD_LIBRARY_PATH //tensorflow/tools/pip_package:build_pip_package

安装：
方式1:

$ pip3 install ~/Downloads/tensorflow-1.8.0-cp36-cp36m-macosx_10_13_x86_64.whl

方式2:

$ pip3 install https://github.com/SixQuant/tensorflow-macos-gpu/releases/download/v1.8.0/tensorflow-1.8.0-cp36-cp36m-macosx_10_13_x86_64.whl

确认：

• 环境变量确认：
  测试脚本：

import os

print(os.environ["DYLD_LIBRARY_PATH"])

$ python3 tensorflow-gpu-01-env.py
/usr/local/cuda/lib:/usr/local/cuda/extras/CUPTI/lib

• GPU启用确认：
  测试脚本：

import tensorflow as tf  
config = tf.ConfigProto()
config.log_device_placement = True
hello = tf.constant('Hello, TensorFlow!')
with tf.Session(config=config) as sess:
    print(sess.run(hello))

出现以下语句表示成功

4. 问题&注意
这里提供一个tf—gpu1.4安装的方案，转自：https://blog.csdn.net/wz22881916/article/details/78807993
推荐👍关于1.8的安装主要参考：

[https://segmentfault.com/a/1190000015807229?utm_source=tag-newest]

注意⚠️：编译配置tf时，一定要连接外接显示器或者插入显卡欺骗器！

参考资料

1. https://segmentfault.com/a/1190000015807229?utm_source=tag-newest
2. https://blog.csdn.net/wz22881916/article/details/78807993
3. https://blog.csdn.net/u014475479/article/details/81702392