用户登录
用户注册

分享至

tensorflow gpu 编译

  • 作者: 太子VS汰渍
  • 来源: 51数据库
  • 2020-10-05
Tensor-flow 默认使用一个gpu进行运算(如果有gpu)。
具体的调用会在程序里。
如果要使用多gpu要相应修改程序。
至于哪部份会放到gpu上跑,一般来说是model training的部分,因为这部分程序是computation intense。
可以把其他部分放到gpu上运行,但是由于cpu-gpu的通信时间较长,会使得程序执行的更慢。



  基本使用
使用 tensorflow, 你必须明白 tensorflow:

使用图 (graph) 来表示计算任务.
在被称之为 会话 (session) 的上下文 (context) 中执行图.
使用 tensor 表示数据.
通过 变量 (variable) 维护状态.
使用 feed 和 fetch 可以为任意的操作(arbitrary operation) 赋值或者从其中获取数据.
软件
前端设计
程序设计
Java相关