Windows上安装TensorFlow无坑教程
手把手在Windows上安装TensorFlow无坑教程
TensorFlow是谷歌基于DistBelief进行研发的第二代人工智能学习系统,可被用于语音或图像识别等多项机器深度学习领域。对于这种高大上的东西估计很多人都想学吧,包括我这位零基础的小编。但网上找到的教程大多都不是新手向的,而且基本用的也不是Windows系统,这让我们这种新手很苦恼。但我参考了Andrew Che的教程,加上实践操作,我成功的在Windows上安装了TensorFlow,以下是我的步骤:
1.检查GPU是否支持CUDN:其实这一步是可以省略的,因为CUDN支持的显卡范围特别广,你基本只要确认你的GPU是不是NVIDIA的就可以了,当然如果你还是不确定的话,也可以去这个网站(https://developer.nvidia.com/cuda-gpus)中的“CUDA-Enabled GeForce Products”查一下,只要你的显卡型号在里面就可以了。
2.下载并按照CUDA:进入此网站(https://developer.nvidia.com/cuda-downloads),点击Windows
选择你的电脑的系统版本(这里我用的是win7)
点击第一个选项
选择第一个下载
下载完之后打开安装,一路确认到底
安装完后进入下一步
3.下载cuDNN:其实,到了这一步比较闹心,因为当你进了下载网站(https://developer.nvidia.com/rdp/form/cudnn-download-survey),你会发现他要让你注册了才能下载,但为了能安装,注册什么的就干了吧,不过程比较烦
然后你就可以现在下载了。考虑到兼容性,这里我们就选择第三个进行下载安装
下载后将其解压缩
在C盘根目录下新建个tools文件夹,将解压后的“cuda”文件夹放入其中
4.设置环境变量:右键“我的电脑”,选择“高级系统设置”
进入“环境变量”
进入“Path”,添加"c:\tools\cuda\bin"与”C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin”后点击确认
5.下载并安装Anaconda:进入下载网站(https://www.continuum.io/downloads),点击Windows图标
这里我们选择版本为3.6的进行下载并安装
下载后打开进行安装,安装时这里选择第一个,也就是它推荐的选项
然后将第两个选项都点上
安装完后进入下一步
6.设置TensorFlow工作环境:在你喜欢的地方新建个tensorflow目录。这里我就在c盘更目录建立了。
按win+r打开“运行”输入cmd,打开命令提示符
输入cd +tensorflow所在目录,回车
接下来依次输入一下几个命令,注意结尾不能有空格,如果途中让你选择是Yes/no,全部输入y
conda create -name tensorflow-gpu python=3.5 anaconda
activate tensorflow-gpu
pip install tensorflow-gpu
pip install keras
转载注: 此处的几个命令有坑,更具体的步骤参考我的一篇博客:http://mp.blog.****.net/postedit/79358543
7.确认TensorFlow安装完成:在命令提示符中输入“jupyter notebook”,在浏览器中召唤jupyter notebook
新建个python3档案
在里面输入以下指令
import tensorflow as tf
tf.__version__
按下shift+enter运行后假如看到1.2.0,说明你成功了,接下来你就可以自由用tensorflow了