美文网首页我爱编程
osx苹果电脑安装TensorFlow(GPU)

osx苹果电脑安装TensorFlow(GPU)

作者: jke_zq | 来源:发表于2017-03-29 22:23 被阅读0次

    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'

    相关文章

      网友评论

        本文标题:osx苹果电脑安装TensorFlow(GPU)

        本文链接:https://www.haomeiwen.com/subject/mmasottx.html