前言:TensorFlow是谷歌基于DistBelief进行研发的第二代人工智能学习系统,CPU/GPU资源对神经网络模型的训练很重要,应充分利用电脑的显卡资源,加快模型的训练速度。
下面是对安装过程的详细讲解:
1、安装好Python的开发环境(可参考我之前的文章https://blog.csdn.net/qq_28821995/article/details/81006704)我这里是在安装好anaconda环境下再来安装TensorFlow的。
2、如果只想安装TensorFlow的CPU版本,可直接使用pip指令,在cmd窗口中输入pip install tensorflow-cpu即可。
3、下面是安装GPU版本的,稍微比较复杂。因为我们是在anaconda下安装,首先激活anaconda,在cmd窗口输入activate Anaconda3进入该环境后继续输入pip install tensorflow-gpu 就能安装GPU版本了。
4、安装GPU版本后,还要继续安装CUDA 各版本的下载链接: