tensorflow是谷歌所开发的用于深度学习,并完全开源的一个库,我也想通过学习tensorflow来逐渐了解深度学习以及机器学习。和以往学过的内容一样,第一步要做的就是配环境。

tensorflow版本详情

  tensorflow在今年支持windows上的安装使用了,如今最新为1.3版本,可能还不是很通用,更多使用的是1.1和1.2版本,根据官网所说,仅支持python3.5版本,不过我使用的python3.6也安装成功了,应该是3.5以后版本都能用。在windows上使用tensorflow,还分为CPU和GPU两版,GPU需要NVIDIA显卡支持,同时安装也更麻烦,不够考虑到CPU的运算速度,还是决定安装GPU版本。

安装所需环境

为成功安装tensorflow的GPU版本,一共需要在电脑安装:

  1. python3.5或以后版本
  2. pip3
  3. VS2015(官网现在没这条,先装了总没错)
  4. CUDA8.0
  5. cuDNNv5.1

安装(踩坑)详情

以上所需软件也可在官网下载安装,需要注意的有,python安装时需要添加PATH变量以便使用pip快速导入包。然后可使用pip指令快速安装tensorflow库
pip3 install --upgrade tensorflow-gpu
虽然不知道为什么,我一开始下载的时候一直报错,可能是网络不稳定吧,总之最后还是能安装的。
VS2015在线安装可能会失败,最好先翻墙。
主要的困难在于CUDA和cuDNN,CUDA安装完后可在命令行输入指令查看
nvcc -V
cuDNN下载后需要将文件添加入PATH环境变量,也可将解压后文件夹中的三个子文件夹中的文件分别复制添加至cuda所安装位置下的文件夹,默认安装路径是:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0,其中对应的bin,include,x64文件夹中。

测试代码

安装完成后可在cmd指令界面打开python,输入以下代码进行测试:

1
2
3
4
5
6
import tensorflow as tf
a = tf.random_normal((10, 10))
b = tf.random_normal((10, 5))
c = tf.matmul(a, b)
sess = tf.InteractiveSession()
sess.run(c)

能跑就基本是安装成功了

示例结果
示例结果