首页 > TensorFlow 阅读:2,083

TensorFlow GPU的安装和使用

本节将学习如何在 GPU 中使用 TensorFlow,所执行的运算操作是在 CPU 或 GPU 上进行简单的矩阵乘法。

首先安装一个支持 GPU 的 TensorFlow 版本。官方的 TensorFlow 安装说明能够提供良好的开始:https://www.tensorflow.org/。请记住,对 GPU 的环境支持需要通过 CUDA 或 CuDNN。

GPU上TensorFlow的使用

  1. 导入几个模块:


     
  2. 从命令行获得想要使用的处理器类型(GPU 或 CPU):


     
  3. 在 GPU 或 CPU 上执行矩阵乘法。关键指令是使用 tf.device(device_name),它创建一个新的上下文管理器来告诉 TensorFlow 在 GPU 或 CPU 上执行这些操作:


     
  4. 打印一些调试计时,以验证 CPU 和 GPU 之间的区别:


此案例说明了如何将 TensorFlow 计算分配给 CPU 或 GPU,代码非常简单,并将作为基础案例。

爱面试的程序媛,一个分享面试经验的公众号。跟着站长一起学习,每天都有进步。

通俗易懂,深入浅出,定时分享程序员面试的那点事。

面试如何造火箭?工作如何拧螺丝?都在这个公号哦。

扫描二维码关注公众号,免费领取价值 1000 元的求职面试资料(限时免费)!

当你决定关注「爱面试的程序媛」,你已然超越了90%的程序员!

爱面试的程序媛二维码
微信扫描二维码关注