github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-07 13:32:09 -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

4
nnlib/nnlib.py
View file

@ -148,13 +148,13 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
import tensorflow as tf import tensorflow as tf
nnlib.tf = tf nnlib.tf = tf
if device_config.cpu_only: if device_config.cpu_only:
config = tf.ConfigProto(device_count={'GPU': 0}) config = tf.ConfigProto(device_count={'GPU': 0})
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 = ''