Device Query: show diagnostic information for a GPU device.
| 135 | |
| 136 | // Device Query: show diagnostic information for a GPU device. |
| 137 | int device_query() { |
| 138 | LOG(INFO) << "Querying GPUs " << FLAGS_gpu; |
| 139 | vector<int> gpus; |
| 140 | get_gpus(&gpus); |
| 141 | for (int i = 0; i < gpus.size(); ++i) { |
| 142 | caffe::Caffe::SetDevice(gpus[i]); |
| 143 | caffe::Caffe::DeviceQuery(); |
| 144 | } |
| 145 | return 0; |
| 146 | } |
| 147 | RegisterBrewFunction(device_query); |
| 148 | |
| 149 | // Load the weights from the specified caffemodel(s) into the train and |