From: James Thewlis Date: Wed, 19 Mar 2014 14:55:44 +0000 (+0000) Subject: Remove cudaSetDevice(1) X-Git-Tag: submit/tizen/20180823.020014~737^2 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=28d27ee908db1f0c78561273f1da3124289224bb;p=platform%2Fupstream%2Fcaffeonacl.git 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 --- 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);