github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-16 10:03:41 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

basic hack for osx

Browse source
This commit is contained in:
jkennedyvz 2022-05-22 18:02:03 -07:00
parent 0d19d8ec8e
commit 8aadbfef4a
3 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
core/leras/device.py
View file

@ -46,7 +46,7 @@ class Devices(object):
idx_mem = 0
for device in self.devices:
mem = device.total_mem
if mem > idx_mem:
if mem >= idx_mem:
result = device
idx_mem = mem
return result
@ -56,7 +56,7 @@ class Devices(object):
idx_mem = sys.maxsize
for device in self.devices:
mem = device.total_mem
if mem < idx_mem:
if mem <= idx_mem:
result = device
idx_mem = mem
return result
@ -270,4 +270,4 @@ class Devices(object):
os.environ[f'NN_DEVICE_{i}_TOTAL_MEM'] = str(device['total_mem'])
os.environ[f'NN_DEVICE_{i}_FREE_MEM'] = str(device['free_mem'])
os.environ[f'NN_DEVICE_{i}_CC'] = str(device['cc'])
"""
"""

1
core/leras/nn.py
View file

@ -112,6 +112,7 @@ class nn():
config.gpu_options.force_gpu_compatible = True
config.gpu_options.allow_growth = True
config.allow_soft_placement = True
nn.tf_sess_config = config
if nn.tf_sess is None: