nnlib initialization: fix choosing best gpu when indexes of gpu are different in tf and nvsmi

2019-09-27 18:47:18 +04:00
parent 4a2203cc35
commit 8041ee959a
1 changed files with 2 additions and 2 deletions
@@ -154,7 +154,7 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
        else:
            config = tf.ConfigProto()

-            if device_config.backend != "tensorflow-generic":
+            if device_config.force_gpu_idx != -1 and device_config.backend != "tensorflow-generic":
                #tensorflow-generic is system with NVIDIA card, but w/o NVSMI
                #so dont hide devices and let tensorflow to choose best card
                visible_device_list = ''