github/DeepFaceLive
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLive synced 2025-07-05 20:42:12 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix WindowsBuilder.py

Browse source
This commit is contained in:
iperov 2022-08-24 22:41:50 +04:00
parent 87f2e30214
commit 3b95be43e0
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
build/windows/WindowsBuilder.py
View file

@ -481,11 +481,11 @@ def build_deepfacelive_windows(release_dir, cache_dir, python_ver='3.7.9', backe
builder.install_pip_package('onnx==1.12.0')
if backend == 'cuda':
builder.install_pip_package('torch==1.10.0+cu113 torchvision==0.11.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html')
#builder.install_pip_package('torch==1.10.0+cu113 torchvision==0.11.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html')
#builder.install_pip_package('torch==1.11.0+cu115 torchvision==0.12.0+cu115 -f https://download.pytorch.org/whl/torch_stable.html')
#builder.install_pip_package('torch==1.12.1+cu116 torchvision==0.13.1+cu116 -f https://download.pytorch.org/whl/torch_stable.html')
builder.install_pip_package('torch==1.12.1+cu116 torchvision==0.13.1+cu116 -f https://download.pytorch.org/whl/torch_stable.html')
builder.install_pip_package('onnxruntime-gpu==1.12.1')
builder.install_pip_package('onnxruntime-gpu==1.11.0')
elif backend == 'directml':
builder.install_pip_package('onnxruntime-directml==1.12.1')
@ -508,6 +508,11 @@ def build_deepfacelive_windows(release_dir, cache_dir, python_ver='3.7.9', backe
print (f'Moving {target}')
shutil.move (str(file), str(target) )
for file in torch_lib_path.glob("**/zlibwapi.dll"):
target = cuda_bin_path / file.name
print (f'Copying {target}')
shutil.copy (str(file), str(target) )
deepfacelive_path = builder.get_internal_path() / 'DeepFaceLive'
print('Copying DeepFaceLive repository.')