Remove cudaSetDevice(1)

So it works when there is only one GPU

diff --git a/tools/dump_network.cpp b/tools/dump_network.cpp
index 286de48..21d0256 100644 (file)
--- a/tools/dump_network.cpp
+++ b/tools/dump_network.cpp
@@ -28,7 +28,6 @@
 using namespace caffe;  // NOLINT(build/namespaces)
 
 int main(int argc, char** argv) {
-  cudaSetDevice(1);
   Caffe::set_mode(Caffe::GPU);
   Caffe::set_phase(Caffe::TEST);