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是相反的,需要特别注意。
本文详细介绍了TensorFlow的CPU和GPU版本的区别,包括安装命令、数据处理格式(NHWC与NCHW),以及在不同模式下卷积运算的实现方式。特别强调了输入张量格式对CPU和GPU模式的影响。

639

被折叠的 条评论
为什么被折叠?



