mirror of
https://github.com/iperov/DeepFaceLive
synced 2025-08-13 18:27:01 -07:00
update onnxruntime
This commit is contained in:
parent
38f015d28e
commit
cea06e4f3a
1 changed files with 2 additions and 2 deletions
|
@ -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()
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue