From 28d27ee908db1f0c78561273f1da3124289224bb Mon Sep 17 00:00:00 2001 From: James Thewlis Date: Wed, 19 Mar 2014 14:55:44 +0000 Subject: [PATCH] Remove cudaSetDevice(1) So it works when there is only one GPU --- tools/dump_network.cpp | 1 - 1 file changed, 1 deletion(-) diff --git a/tools/dump_network.cpp b/tools/dump_network.cpp index 286de485..21d0256f 100644 --- 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);