diff --git a/AIEngineModule/include_base/AI_Factory.h b/AIEngineModule/include_base/AI_Factory.h index de005b8..940ca8b 100644 --- a/AIEngineModule/include_base/AI_Factory.h +++ b/AIEngineModule/include_base/AI_Factory.h @@ -41,7 +41,7 @@ struct GPU_Config UseGPUList[i] = -1; } UseGPUList[0] = 0; - UseGPUList[1] = 1; + UseGPUList[1] = -1; } int GetNum() { diff --git a/AlgorithmModule/include/ImgCheckBase.h b/AlgorithmModule/include/ImgCheckBase.h index 0f64959..c0dedfa 100644 --- a/AlgorithmModule/include/ImgCheckBase.h +++ b/AlgorithmModule/include/ImgCheckBase.h @@ -3,7 +3,7 @@ #include #include #define ALL_INTERFACE_VERSION 7 -#define MAX_GPU_NUM 4 +#define MAX_GPU_NUM 1 enum CHECK_THREAD_RUN_STATUS { @@ -53,7 +53,7 @@ struct RunInfoST UseGPUList[i] = -1; } UseGPUList[0] = 0; - UseGPUList[1] = 1; + UseGPUList[1] = -1; } void copy(RunInfoST tem) { diff --git a/AlgorithmModule/src/ALLImgCheckAnalysisy.cpp b/AlgorithmModule/src/ALLImgCheckAnalysisy.cpp index 9319eff..60130d7 100644 --- a/AlgorithmModule/src/ALLImgCheckAnalysisy.cpp +++ b/AlgorithmModule/src/ALLImgCheckAnalysisy.cpp @@ -329,7 +329,7 @@ int ALLImgCheckAnalysisy::InitAIFactory() if (num == 0) { gpu.UseGPUList[0] = 0; - gpu.UseGPUList[1] = 1; + gpu.UseGPUList[1] = -1; } for (int i = 0; i < MAX_GPU_NUM; i++) {