PaddlePaddle / Mobile

Embedded and Mobile Deployment

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Android初始化PaddlePaddle问题

yeyupiaoling opened this issue · comments

看Android得例子,初始化代码应该是这一段

void ImageRecognizer::init_paddle() {
static bool called = false;
if (!called) {
// Initalize Paddle
char* argv[] = {const_cast<char*>("--use_gpu=False"),
const_cast<char*>("--pool_limit_size=0")};
CHECK(paddle_init(2, (char**)argv));
called = true;
}
}

其中--pool_limit_size=0是什么意思,是否可以在初始化PaddlePaddle的时候指定使用使用GPU
我看到代码指定是否使用GPU是这一段

paddle_matrix mat = paddle_matrix_create(
/* sample_num */ 1,
/* size */ normed_channel_ * normed_height_ * normed_width_,
/* useGPU */ false);

应都是指定GPU的,只是如果初始化的时候,没有GPU,但指定CPU的时,初始化PaddlePaddle就报错了。如果是使用创建矩阵的时候指定GPU,只是不能正常预测,其他都是正常的。