我爱编程

Win10搭建TensorFlow(r1.5) CUDA9.0&

2017-06-28  本文已影响327人  shohokuooo

最近因为工作需要,要研究下TensorFlow,所以总结下我这两天在Windows平台上安装的经历。或许有人能用到。
首先说下,我安装的时候TensorFlow的最新版本是1.5,所以接下来说的都是这个版本的情况,这个非常非常重要,无论如何官网都是第一要参考的,其次才是别人的总结。

准备工作

打开官网:Getting Started With TensorFlow进入Install链接
按照上面说的,有两种方式来使用TensorFlow,一种是CPU模式,另一种是GPU模式。我也按照这两种模式说下我的做法。

补充升级到TensorFlow r1.3版本的GPU模式安装的问题,在r1.2版本中使用的是cuDNN v5.1,在r1.3使用的是cuDNN v6 or v6.1,所以需要重新下载并替换掉之前的v5.1版本。然后还是下面的安装步骤就没问题了。不需要卸载,直接安装就是升级。*In particular, the cuDNN version must match exactly: TensorFlow will not load if it cannot find cuDNN64_6.dll. *

开始安装

conda create -n tensorflow python=3.6 anaconda

我在后面加了个anaconda,推荐大家也加上,这样np,pd,sklearn,scipy之类的包就都一起安装了。
之后activate到新的环境上继续下面的步骤

pip install --ignore-installed --upgrade tensorflow 
pip install --ignore-installed --upgrade tensorflow-gpu 

然后接下来就是import tensorflow,并print Hello TensorFlow!了

Pycharm环境

在Pycharm中使用TensorFlow也很简单。
首先是创建一个准备使用TensorFlow的Project,然后为其建立一个Interpreter:

Add local 选择Anaconda2\envs\tensorflow下的python.exe 搞定Interpreter

然后就可以运行Hello代码了。

import tensorflow as tf

hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()

print(sess.run(hello))
运行结果

可以发现有些告警,是因为我们使用的预编译库没有SSE等指令,这个可以给CPU加速1倍左右。不过需要我们重新编译生成库,不过因为Tensor非常不推荐在Windows做这个事情,所以想重新编译的话还是在Ubuntu上做吧,下个最新版的编译起来也不费劲。因为我主要是想单机做些简单实验,而且我使用的是GPU模式,所以就不在win10上折腾了。
至此简单的TensorFlow环境就搭建好了,接下来就可以开始学习Tensor、Graph、Node等等新的知识了。

上一篇 下一篇

猜你喜欢

热点阅读