Skip to content
This repository has been archived by the owner on Nov 13, 2024. It is now read-only.

Commit

Permalink
update onnxruntime
Browse files Browse the repository at this point in the history
  • Loading branch information
@iperov
iperov committed Nov 8, 2022
1 parent 38f015d commit cea06e4
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions build/windows/WindowsBuilder.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -485,9 +485,9 @@ def build_deepfacelive_windows(release_dir, cache_dir, python_ver='3.7.9', backe
#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('onnxruntime-gpu==1.11.0')
builder.install_pip_package('onnxruntime-gpu==1.13.1')
elif backend == 'directml':
builder.install_pip_package('onnxruntime-directml==1.12.1')
builder.install_pip_package('onnxruntime-directml==1.13.1')

builder.install_ffmpeg_binaries()

Expand Down

0 comments on commit cea06e4

Please sign in to comment.