Fix OpenCV compilation. maxpool_layer on GPU uses cuDNN.

This commit is contained in:
AlexeyAB
2018-10-10 13:51:36 +03:00
parent dbc75aec77
commit 7ee4135910
6 changed files with 55 additions and 3 deletions

View File

@ -370,6 +370,8 @@ int main(int argc, char **argv)
gpu_index = find_int_arg(argc, argv, "-i", 0);
if(find_arg(argc, argv, "-nogpu")) {
gpu_index = -1;
printf("\n Currently Darknet doesn't support -nogpu flag. If you want to use CPU - please compile Darknet with GPU=0 in the Makefile, or compile darknet_no_gpu.sln on Windows.\n");
exit(-1);
}
#ifndef GPU