tensorflow安装的坑

本文详细介绍了TensorFlow的CPU和GPU版本的区别,包括安装命令、数据处理格式(NHWC与NCHW),以及在不同模式下卷积运算的实现方式。特别强调了输入张量格式对CPU和GPU模式的影响。

1.默认安装的是cpu版本的

      conda   install   tensorflow==1.7.0

      cpu版本的tensorflow只支持NHWC的数据读取或者处理格式

2.如果想要安装gpu版本的,需要指定

      conda   install  tensorflow-gpu==1.7.0

      gpu版本的tensorflow可以支持两种,但是默认状态是NCHW

     具体关于通道的解释可以参考https://blog.csdn.net/zdy0_2004/article/details/60871183

开始卷积计算。Conv2DOp实现了CPU和GPU两种模式下的卷积运算。同时,还需要注意input tensor的输入格式,通常有NHWC和NCHW两种格式。在TF中,Conv2d-CPU模式下目前仅支持NHWC格式,即[Number, Height, Weight, Channel]格式。Conv2d-GPU模式下以NCHW为主,但支持将NHWC转换为NCHW求解。C++中多维数组是row-major顺序存储的,而Eigen默认是col-major顺序的,则C++中[N, H, W, C]相当于Eigen中的[C, W, H, N],即dimention order是相反的,需要特别注意。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值