深度学习初学者指南
上QQ阅读APP看书,第一时间看更新

2.2.2 拥有GPU支持的TensorFlow

在默认情况下,Colaboratory会自动为TensorFlow启用GPU支持。然而,如果你自己的系统有一个GPU,并想安装支持GPU的TensorFlow,安装是非常简单的。只需要在你的个人系统上输入下列命令:

但是,请注意,这假定你已经为系统安装了所有必要的驱动程序来访问GPU。然而,不要害怕,可以在互联网上搜索到很多关于这个过程的文档,例如,https://www.tensorflow.org/install/gpu。如果你遇到任何问题,并且需要继续推进,我强烈建议你回到Colaboratory,因为这是最简单的学习方法。

现在,我们来讨论一下TensorFlow的工作原理,以及它的图形范式是如何使它具有非常强的健壮性的。