Remove cudaSetDevice(1) · 28d27ee908 - caffe - Git

Remove cudaSetDevice(1)

Просмотр исходного кода

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 удалений

									
										1

tools/dump_network.cpp

										Просмотреть файл
									
					@ -28,7 +28,6 @@

					using namespace caffe;  // NOLINT(build/namespaces)

					using namespace caffe;  // NOLINT(build/namespaces)

					int main(int argc, char** argv) {

					int main(int argc, char** argv) {

					  cudaSetDevice(1);

					  Caffe::set_mode(Caffe::GPU);

					  Caffe::set_mode(Caffe::GPU);

					  Caffe::set_phase(Caffe::TEST);

					  Caffe::set_phase(Caffe::TEST);