So it works when there is only one GPU
This commit is contained in:
James Thewlis 2014-03-19 14:55:44 +00:00
Родитель 0f9a28dace
Коммит 28d27ee908
1 изменённых файлов: 0 добавлений и 1 удалений

Просмотреть файл

@ -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);