github/DeepFaceLive
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLive synced 2025-07-13 16:43:52 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

WindowsBuilder.py: upgrade to onnxruntime-gpu==1.9.0 for CUDA build

Browse source
This commit is contained in:
iperov 2021-09-22 23:53:23 +04:00
parent cd00e0ea31
commit cfc672f596
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
build/windows/WindowsBuilder.py
View file

@ -481,7 +481,7 @@ def build_deepfacelive_windows(release_dir, cache_dir, python_ver='3.7.9', backe
if backend == 'cuda':
builder.install_pip_package('torch==1.8.1+cu111 torchvision==0.9.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html')
builder.install_pip_package('onnxruntime-gpu==1.8.1')
builder.install_pip_package('onnxruntime-gpu==1.9.0')
builder.install_pip_package('cupy-cuda111===9.0.0')
elif backend == 'directml':
if python_ver[:3] == '3.7':