зеркало из https://github.com/microsoft/caffe.git
Remove cudaSetDevice(1)
So it works when there is only one GPU
This commit is contained in:
Родитель
0f9a28dace
Коммит
28d27ee908
|
@ -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);
|
||||
|
||||
|
|
Загрузка…
Ссылка в новой задаче