mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-08-19 04:59:27 -07:00
Fixed bug on leras
This commit is contained in:
parent
deae2f19e9
commit
d15fe9ebaf
1 changed files with 1 additions and 1 deletions
|
@ -107,7 +107,7 @@ class nn():
|
|||
else:
|
||||
nn.tf_default_device_name = f'/{device_config.devices[0].tf_dev_type}:0'
|
||||
|
||||
config = tf.ConfigProto()
|
||||
config = tf.ConfigProto(allow_soft_placement=True)
|
||||
config.gpu_options.visible_device_list = ','.join([str(device.index) for device in device_config.devices])
|
||||
|
||||
config.gpu_options.force_gpu_compatible = True
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue