TensorFlow的安装分为支持GPU和不支持GPU两种。但每种安装方式有好多种,这里只介绍最简单的支持GPU的安装。
一、是否支持GPU
查看NVIDIA documentation对应mac上的GPU型号,如果兼容性在3.0以及以上则『是』,否则『否』。
二、检查和升级
1.mac osx是否是10.11以及以上,如果不是,请升级;
2.xcode是否是7.3以及以上,如果不是,请升级;
三、安装NVIDIA相关驱动
1.CUDA Toolkit 8.0. For details, seeNVIDIA's documentation.
step1:从Download下载,并安装;
step2:export PATH="/Developer/NVIDIA/CUDA-8.0.61/bin:$PATH"
step3:export DYLD_LIBRARY_PATH="/Developer/NVIDIA/CUDA-8.0.61/lib:$DYLD_LIBRARY_PATH"
2.cuDNN v5.1. For details, seeNVIDIA's documentation.
step1:从Download下载cuDNN v6.0
step2:tar xzvf /path_to/cudnn-8.0-osx-x64-v5.1.tgz -C /target_path
step3:cd /target_path/cuda
step4:export DYLD_LIBRARY_PATH=`pwd`:$DYLD_LIBRARY_PATH
3.安装TensorFlow
step1:sudo easy_install pip
step2:pip install --upgrade tensorflow-gpu(如果失败,尝试执行step4)
step3:如果step2失败,执行step4
step4:pip install --upgrade https://storage.googleapis.com/tensorflow/mac/gpu/tensorflow_gpu-1.0.1-py2-none-any.whl (如果失败,尝试执行step5)
step5:sudo -H pip install https://storage.googleapis.com/tensorflow/mac/gpu/tensorflow_gpu-1.0.1-py2-none-any.whl --upgrade --ignore-installed
4.关掉SIP(Security Integrity Protection)
step1:按住comand+R,同时按住电源键;当出现苹果logo的时候,松开电源键
step2:等待出现系统工具窗口
step3:选择工具-->终端,然后输入:csrutil disable
step4:选择苹果图标-->重启
5.验证TensorFlow是否安装成功
终端输入:python -c 'import tensorflow as tf'