美文网首页程序员
TensorFlow的安装(MAC系统CPU版本 & Windo

TensorFlow的安装(MAC系统CPU版本 & Windo

作者: 东方耀888 | 来源:发表于2018-09-04 10:35 被阅读0次

    TensorFlow的安装(MAC系统CPU版本 & Windows系统GPU版本)

    首先,TensorFlow的安装要求Python必须是64位的,如图

    TensorFlow的安装分为CPU版本与GPU版本:

    CPU版本:程序可以在CPU上计算或运行,默认在CPU上,所有的计算机都有CPU

    GPU版本:程序既可以在CPU上也可以在GPU上计算或运行,默认是在GPU上运行,可以指定程序在CPU上运行,GPU上的运算速度比CPU快,所以一般条件满足的话尽量安装GPU的版本

    安装GPU版本的前提条件:需要机器的显卡必须是NVIDIA的显卡

    对于Python3.5或Python3.6,可以使用pip install tensorflow或pip install tensorflow-gpu直接安装,对于python2.7版本的Windows系统,只能通过源码编译来安装,其他系统比如Mac或Linux也可以直接通过命令来安装

    CPU版本的安装可以通过指定TensorFlow版本的形式安装,例如:

    Pip install tensorflow==1.8.0

    安装成功之后通过pip list可以看到tensorflow与tensorflow-tensorboard就代表已经安装上了,但是还不一定成功,我们需要通过在python环境下import tensorflow来测试,如果没有异常就证明安装成功了,如果有异常还需要进一步的处理

    CPU版本的安装相对比较简单,如果是第一次安装GPU版本的话,此时就会出现异常

    根据提示,需要我们下载CUDA(这是NVIDIA提供给程序使用GPU的接口),毕竟是使用人家的东西,所以需要专门的接口。

    CUDA下载安装链接:https://developer.nvidia.com/cuda-toolkit-archive

    CUDA这里有很多的版本,具体的选择根据异常的提示来,比如上文的例子就是需要下载8.0版本的CUDA

    之后,我们还需要下载一个cuDNN,链接为:https://developer.nvidia.com/rdp/cudnn-archive

    cuDNN版本的选择与CUDA版本对应即可

    CUDA的安装步骤大概如下(默认路径,最后选择精简):

    cuDNN的文件下载后直接解压,将里面的三个文件夹直接覆盖到CUDA的安装路径下即可

    最后,我们再次import tensorflow就不会出现异常了,我们就成功啦!

    如果文字版的教程大家还是不太明白,我也录制了对应的视频版,里面讲得更加详细,干货更多,每一步的操作大家都可以看得明明白白,欢迎下载!

    04、TensorFlow的安装CPU版本_MAC系统:下载链接:https://pan.baidu.com/s/11-tjRr8SQPGGfURGLAW7oQ 密码:ko7r


    04、TensorFlow的安装CPU版本_MAC系统

    05、TensorFlow的安装GPU版本_Windows系统:下载链接:https://pan.baidu.com/s/12xTGJUhEgSMPyjflkB_Jxw 密码:rjjr


    05、TensorFlow的安装GPU版本_Windows系统

    相关文章

      网友评论

        本文标题:TensorFlow的安装(MAC系统CPU版本 & Windo

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