diff --git a/build/windows/WindowsBuilder.py b/build/windows/WindowsBuilder.py index 9d48e8e..a538e38 100644 --- a/build/windows/WindowsBuilder.py +++ b/build/windows/WindowsBuilder.py @@ -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':