in Caffe::SetDevice, call cudaSetDevice before Get
authorJonathan L Long <jonlong@cs.berkeley.edu>
Mon, 16 Jun 2014 22:20:17 +0000 (15:20 -0700)
committerJonathan L Long <jonlong@cs.berkeley.edu>
Mon, 16 Jun 2014 22:20:17 +0000 (15:20 -0700)
commit47ef5629d8ad73aba4a262612ea4c3f537b08c17
tree423c563763ac5c2c568f0c8193024cecd55a3c6e
parent6b6ab3339f61c6edf91e23878b23b22e18c16398
in Caffe::SetDevice, call cudaSetDevice before Get

Otherwise initialization will be performed on whichever device is
default.
src/caffe/common.cpp