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

fixed GPU indexing

Browse Source
This commit is contained in:
Colombo
2019-10-10 13:13:36 +04:00
parent d2d0dcb5e7
commit 59ad734b6c
1 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
nnlib/nnlib.py
View File

@@ -157,13 +157,10 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
else:
config = tf.ConfigProto()
#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 = ''
# for idx in device_config.gpu_idxs:
# visible_device_list += str(idx) + ','
# config.gpu_options.visible_device_list=visible_device_list[:-1]
visible_device_list = ''
for idx in device_config.gpu_idxs:
visible_device_list += str(idx) + ','
config.gpu_options.visible_device_list=visible_device_list[:-1]
config.gpu_options.force_gpu_compatible = True
config.gpu_options.allow_growth = device_config.allow_growth