我爱编程

windows环境下安装TensorFlow(安装篇)

2017-09-07  本文已影响750人  小纠结在简书

TensorFlow是谷歌基于DistBelief进行研发的第二代人工智能学习系统,TensorFlow可被用于语音识别或图像识别等多项机器深度学习领域,是可以快速用于二次开发的人工智能学习系统。

以下为本人在学习TensorFlow过程中windows环境安装步骤,希望能对初学者有所帮助。

一、安装python3.5.x版本

1、地址:https://www.python.org/downloads/

2、注意:请下载3.5.X相关版本,因为windows版本tensorfow暂时只兼容3.5.X版本python

二、更新pip

1、命令:python -m pip install --upgrade pip

注:以上命令直接在window命令行中执行

image.png

三、安装tensorflow

1、CPU版本:pip3 install --upgrade tensorflow

2、GPU版本:pip3 install --upgrade tensorflow-gpu

注:如果你的显卡支持GPU,请安装GPU版本,如果不支持安装CPU版本即可。

image.png

四、安装CUDA

1、地址:https://developer.nvidia.com/cuda-downloads

注:只在GPU版本需要安装

五、安装cudnn

1、地址:https://developer.nvidia.com/rdp/cudnn-download

2、操作:把E:downloadcudnn-8.0-windows10-x64-v5.1cuda下的所有文件复制到C:Program FilesNVIDIA GPU Computing ToolkitCUDA�8.0

注:只在GPU版本需要安装

六、安装pycharm IDE

1、地址:https://www.jetbrains.com/pycharm/download/download-thanks.html?platform=windows

注:根据个人习惯选择不同的开发IDE即可

说明:详情可参数https://www.tensorflow.org/install/

上一篇下一篇

猜你喜欢

热点阅读