沧海一粟

More know, more less

Tensorflow 源码分析-  从GPU OOM开始说Tensorflow的BFC内存管理

前言在平台上跑GPU训练,结果CUDA OOM了,错误提示E Internal: failed initializing StreamExecutor for CUDA device ordinal 0: Internal: failed call to cuDevicePrimary CtxR...

2018-05-04 11:46:13

阅读数:2040

评论数:0

Tensorflow 源码分析-会话与线程池之间的关系

1. Tensorflow 的sessionFactory创建新的会话,tensorflow使用了多工厂模式,在不同的场景下使用不同的工厂, 使用什么工厂模式由传递进来的SessionOptions来决定。1.1 注册工厂Tensorflow 提供了可以注册多会话工厂的模式,允许不同的模块注册自己...

2018-04-02 17:11:16

阅读数:430

评论数:0

tf.transpose 函数说明和用法

tf.transpose(input, [dimension_1, dimenaion_2,..,dimension_n])说明:函数用于交换输入张量维度参数:input 输入张量dimension_n 张量的纬度,输入整数如果输入张量是二维用[0,1]表示,[1,0]就相当于转置矩阵。如果张量是...

2018-01-23 14:58:12

阅读数:464

评论数:0

Tensorflow 源码分析-GPU调用是如何实现的

1. Tensorflow GPU支持Tensorflow 支持GPU进行运算,目前官方版本只支持NVIDIA的GPU,可以在tensorflow的官方上看到。Tensorflow 对GPU的运算的支持最小力度就是OP,也就是我们常说的算子,下图提供了Tensorflow的一些常见算子,而每个算子...

2017-12-21 10:22:30

阅读数:3206

评论数:0

提示
确定要删除当前文章?
取消 删除