Win10下Tensorflow的安装

Win10下Tensorflow的安装

1. Tensorflow简介

TensorFlow是谷歌基于DistBelief进行研发的第二代人工智能学习系统,其命名来源于本身的运行原理。Tensor(张量)意味着N维数组,Flow(流)意味着基于数据流图的计算,TensorFlow为张量从流图的一端流动到另一端计算过程。TensorFlow是将复杂的数据结构传输至人工智能神经网中进行分析和处理过程的系统1

目前官方发布的稳定版本为1.10。

2. 安装

安装之前,首先需要确定安装CPU版还是GPU版,以下过程以GPU版为例。

安装GPU版本的tensorflow,首先得安装以下NVIDIA软件:

  1. CUDA工具包9.0;
  2. 与CUDA9.0相关的驱动程序;
  3. cuDNN7.0;

注意: 需要CUDA 计算能力为 3.5 或更高的 GPU 卡,以及CUDA、cuDNN的版本一定要一致。

tensorflow官网上提供了两种安装方式,分别是:

  1. "原生"pip
  2. Anaconda

以下只介绍pip安装。

在完成上述NVIDIA软件之后,在命令行窗口中,输入以下命令,并回车。

pip install tensorflow-gpu

进度条完成,即完成了tensorflow的安装。

3. 测试

完成安装以后,需要测试是否正确安装。

打开命令行窗口,输入以下命令进入python交互界面。

python

然后依次输入(也可以将以下代码复制到python脚本中运行),第一次运行需要的时间可能会比较长一些。

import tensorflow as tf
hello = tf.constant('Hello, Tensorflow!')
sess = tf.Session()
print(sess.run(hello))

如图所示:

Win10下Tensorflow的安装

至此,安装完成。

4. 其他

4.1 安装CUDA

tensorflow 1.10只能用CUDA9.0。在NVIDIA官网下载相应版本,然后根据提示安装即可。点击此处下载。

4.2 安装cuDNN

CUDA安装完成之后,进入安装目录,一般是在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0,将cuDNN解压,并将得到的三个文件夹(bin, include, lib)复制到CUDA的安装目录。

4.3 注意事项

在安装CUDA期间,如果出错,把安全防护软件关闭后再试试。PS:我是不会告诉你,火绒会影响CUDA的安装……

另外就是,目前,2018年9月19日,tensorflow是不支持python3.7的,用python3.6就好。


[1] 百度百科.https://baike.baidu.com/item/TensorFlow/18828108?fr=aladdin

上一篇:[Leetcode][Python]45: Jump Game II


下一篇:[leetcode]Best Time to Buy and Sell Stock II @ Python