github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-07 21:42:08 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

2
nnlib/nnlib.py
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 = ''