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

Bug Fixes: Improved Multi-GPU Support in Windows System for Merge Command.

Browse source
This commit is contained in:
YuMo 2023-04-21 11:53:17 +08:00
parent 9ef04b2207
commit a8c039da93
3 changed files with 11 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

1
core/leras/device.py
View file

@ -68,6 +68,7 @@ class Devices(object):
return None return None
def get_devices_from_index_list(self, idx_list): def get_devices_from_index_list(self, idx_list):
idx_list = [int(idx.strip()) for idx in idx_list.split(',')]
result = [] result = []
for device in self.devices: for device in self.devices:
if device.index in idx_list: if device.index in idx_list:

2
main.py
View file

@ -1,4 +1,5 @@
if __name__ == "__main__": if __name__ == "__main__":
# Fix for linux # Fix for linux
import multiprocessing import multiprocessing
multiprocessing.set_start_method("spawn") multiprocessing.set_start_method("spawn")
@ -149,7 +150,6 @@ if __name__ == "__main__":
p.add_argument('--cpu-only', action="store_true", dest="cpu_only", default=False, help="Train on CPU.") p.add_argument('--cpu-only', action="store_true", dest="cpu_only", default=False, help="Train on CPU.")
p.add_argument('--force-gpu-idxs', dest="force_gpu_idxs", default=None, help="Force to choose GPU indexes separated by comma.") p.add_argument('--force-gpu-idxs', dest="force_gpu_idxs", default=None, help="Force to choose GPU indexes separated by comma.")
p.add_argument('--silent-start', action="store_true", dest="silent_start", default=False, help="Silent start. Automatically chooses Best GPU and last used model.") p.add_argument('--silent-start', action="store_true", dest="silent_start", default=False, help="Silent start. Automatically chooses Best GPU and last used model.")
p.add_argument('--execute-program', dest="execute_program", default=[], action='append', nargs='+') p.add_argument('--execute-program', dest="execute_program", default=[], action='append', nargs='+')
p.set_defaults(func=process_train) p.set_defaults(func=process_train)

4
requirements-cuda.txt
View file

@ -1,8 +1,8 @@
tqdm tqdm
numpy==1.19.3 numpy==1.19.5
numexpr numexpr
h5py==2.10.0 h5py==2.10.0
opencv-python==4.1.0.25 opencv-python==4.1.2.30
ffmpeg-python==0.1.17 ffmpeg-python==0.1.17
scikit-image==0.14.2 scikit-image==0.14.2
scipy==1.4.1 scipy==1.4.1