选择哪块显卡运行ocr_test

[HouBiaoLiu]

我在进行ocr_test测试时，使用第0块显卡；现在我想通过命令行参数传递使用哪块选卡，于是做了如下修改
bool Classifier::Init(const string& model_path, int gpu_id)
{
const string trained_file = model_path + "/model.caffemodel";
const string model_file = model_path + "/deploy.prototxt";
string mean_file = model_path + "/mean.binaryproto";
const string mean_value_file = model_path + "/mean_values.txt";
const string label_file = model_path + "/label.txt";

if (!CheckFileExist(mean_file.c_str()))
mean_file = mean_value_file;
if (-1 == gpu_id)
{
LOG(INFO) << "Use CPU.";
Caffe::set_mode(Caffe::CPU);
}
else
{
LOG(INFO) << "Use GPU with device ID " << gpu_id;
cudaDeviceProp device_prop;
cudaGetDeviceProperties(&device_prop, gpu_id);
LOG(INFO) << "GPU device name: " << device_prop.name;
Caffe::SetDevice(gpu_id);
Caffe::set_mode(Caffe::GPU);
}
重新编译运行时，选择第3块显卡，结果第0块和第三块显卡一起在跑，请问什么原因啊
但是我跑这种./build/tools/caffe test却是正常的