CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Colombo
2019-09-27 18:47:18 +04:00
parent 4a2203cc35
commit 8041ee959a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nnlib/nnlib.py
+1 -1
View File
@@ -154,7 +154,7 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
else: else:
config = tf.ConfigProto() 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 #tensorflow-generic is system with NVIDIA card, but w/o NVSMI
#so dont hide devices and let tensorflow to choose best card #so dont hide devices and let tensorflow to choose best card
visible_device_list = '' visible_device_list = ''