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

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: else:
config = tf.ConfigProto() config = tf.ConfigProto()
#if device_config.force_gpu_idx != -1 and device_config.backend != "tensorflow-generic": visible_device_list = ''
# #tensorflow-generic is system with NVIDIA card, but w/o NVSMI for idx in device_config.gpu_idxs:
# #so dont hide devices and let tensorflow to choose best card visible_device_list += str(idx) + ','
# visible_device_list = '' config.gpu_options.visible_device_list=visible_device_list[:-1]
# 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.force_gpu_compatible = True
config.gpu_options.allow_growth = device_config.allow_growth config.gpu_options.allow_growth = device_config.allow_growth