calculate gpu memory

zhangmeng

2020-01-19 c0fece4c4978cde052889bfcb8e0561342596eca

calculate gpu memory

 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;