我爱编程

osx苹果电脑安装TensorFlow(GPU)

2017-03-29  本文已影响0人  jke_zq

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'

上一篇下一篇

猜你喜欢

热点阅读