csrc/common/gpu/info.cpp
@@ -380,7 +380,7 @@ int mem_idle = need; for(int i = 0; i < gpu_info.device_count; i++){ if (i != lastChoice){ int mem_free = gpu_info.devices[i].memory_free >> 20; int mem_free = (gpu_info.devices[i].memory_free >> 20) - reserved; if(mem_free > mem_idle){ mem_idle = mem_free; suitable_gpu = i;
