天天看点

Windows下如何配置TensorFlow?这有个简单明了的教程(支持GPU哦)

tensorflow是谷歌基于distbelief进行研发的第二代人工智能学习系统,其命名来源于本身的运行原理。tensor(张量)意味着n维数组,flow(流)意味着基于数据流图的计算,tensorflow为张量从流图的一端流动到另一端计算过程。tensorflow是将复杂的数据结构传输至人工智能神经网中进行分析和处理过程的系统。tensorflow可被用于语音识别或图像识别等多项机器深度学习领域。

Windows下如何配置TensorFlow?这有个简单明了的教程(支持GPU哦)

使用gpu进行深度学习被广泛告知为高度有效。显然,非常高端的gpu集群可以通过深度学习做一些惊人的事情。然而,我很好奇windows surface book(gpu:geforce gt 940)使用gpu与cpu的性能对比。事实证明使用gpu比cpu性能高的很多:

更新的surface book拥有更先进的gpu(geforce gt 965)。去年,mac和windows之间的tensorflow领域真的发生大变化。当tensorflow首次发布时没有windows版本,但现在nvidia cuda针对深入学习有着很大兴趣的发展。

安装

首先,你应该确保你已经安装了正确的nvidia驱动程序:

<a href="http://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/">cuda驱动程序</a>

安装tensorflow到windows python,tensorflow需要python2.7、3.4或3.5版本。我使用的是anaconda python3.5。我所做第一件事就是为tensorflow创建cpu和gpu环境。这使他们与我有其他非深入学习python环境分开。创建cpu tensorflow环境:

创建gpu tensorflow环境:

使用单个gpu时,你的tensorflow代码不会被更改。你可以通过切换环境简单地运行相同的代码。tensorflow使用gpu或不使用,这取决于你所处的环境。您可以在以下环境之间切换:

结论

如果你在本地计算机上做中等深度学习网络和数据集,你应该使用你的gpu。即使你正在使用一台笔记本电脑。nvidia是科学计算的首选gpu。虽然amd可能完全有能力,但对amd的支持却很稀少。

文章原标题《using tensorflow in windows with a gpu》,作者:jeff heaton,译者:海棠

翻译者: 海棠 

wechat:269970760 

<a href="https://weibo.com/u/3910993635/" target="_blank">weibo:uncle_lld </a>

email:[email protected]

微信公众号:ai科技时讯

Windows下如何配置TensorFlow?这有个简单明了的教程(支持GPU哦)

继续阅读