Update gpuScheduler.ts (microsoft#1043)

SparkSnail · May 14, 2019 · 34988d9 · 34988d9
1 parent 9974c60
commit 34988d9
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/nni_manager/training_service/local/gpuScheduler.ts b/src/nni_manager/training_service/local/gpuScheduler.ts
@@ -73,8 +73,13 @@ class GPUScheduler {
 
     public getAvailableGPUIndices(): number[] {
         if (this.gpuSummary !== undefined) {
-            return this.gpuSummary.gpuInfos.filter((info: GPUInfo) => info.activeProcessNum === 0)
-                .map((info: GPUInfo) => info.index);
+            if(process.platform === 'win32') {
+                return this.gpuSummary.gpuInfos.map((info: GPUInfo) => info.index);
+            }
+            else{
+                return this.gpuSummary.gpuInfos.filter((info: GPUInfo) => info.activeProcessNum === 0)
+                    .map((info: GPUInfo) => info.index);
+            }
         }
 
         return [];