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
05、TensorFlow的安装GPU版本_Windows系统:下载链接:https://pan.baidu.com/s/12xTGJUhEgSMPyjflkB_Jxw 密码:rjjr