Added device_query.cpp to examples/ to get basic information about th…

…e current GPU device or other device_id

examples/devicequery.cpp

@@ -0,0 +1,23 @@
+// Copyright 2013 Yangqing Jia
+
+
+#include "caffe/common.hpp"
+#include "caffe/net.hpp"
+
+
+using namespace caffe;
+
+int main(int argc, char** argv) {
+  if (argc > 2) {
+    LOG(ERROR) << "device_query [device_id=0]";
+    return 0;
+  }
+  if (argc == 2) {
+    LOG(INFO) << "Querying device_id=" << argv[1];
+    Caffe::SetDevice(atoi(argv[1]));
+    Caffe::DeviceQuery();
+  } else {
+    Caffe::DeviceQuery();
+  }
+  return 0;
+}

src/caffe/common.cpp

@@ -94,6 +94,7 @@ void Caffe::DeviceQuery() {
     return;
   }
   CUDA_CHECK(cudaGetDeviceProperties(&prop, device));
+  printf("Device id:                     %d\n", device);
   printf("Major revision number:         %d\n", prop.major);
   printf("Minor revision number:         %d\n", prop.minor);
   printf("Name:                          %s\n", prop.name);