如何知道CAFFE代码是否正在使用GPU或CUDA代码运行？

Question

我知道，CAFFE中的几乎所有层都可以在CPU / GPU中运行。 对于GPU，它具有某种模式，称为引擎。 如果engine=CAFFE ，它将与GPU一起运行，而engine=CUDNN ，它将基于CUDA代码运行。 默认为默认模式

在我的Makefile.config中，在构建caffe时打开CUDNN模式

# cuDNN acceleration switch (uncomment to build with cuDNN).
USE_CUDNN := 1

在我的原型中，我有一个层，例如反卷积层

layer {
  name: "DeconvolutionLayer"
  type: "Deconvolution"
  bottom: "conv1"
  top: "DeconvolutionLayer"
  param {
    lr_mult: 1
    decay_mult: 1
  }
  convolution_param {
    num_output: 128
    bias_term: false
    pad: 0
    kernel_size: 2
    stride: 2
    weight_filler {
      type: "msra"
    }    
  }
}

请注意，我没有为engine标志设置任何值。 当我运行时，将使用GPU还是CUDNN是哪种模式？ 训练脚本是

caffe/build/tools/caffe train --solver=solver.prototxt -gpu 0

Answer 1

如果查看caffe.proto它会显示"The default for the engine is set by the ENGINE switch at compile-time." 。 如果你看看layer_factory.cpp如果USE_CUDNN设置，那么默认的引擎设置为cuDNN 。 所以您的情况是CUDNN。