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

update onnxruntime versions

Browse source
This commit is contained in:
iperov 2021-12-18 21:59:52 +04:00
parent 77cbb05db4
commit 69f71ddecd
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
build/linux/Dockerfile
View file

@ -9,7 +9,7 @@ RUN ln -s /usr/bin/python3 /usr/bin/python
RUN git clone https://github.com/iperov/DeepFaceLive.git
RUN python -m pip install --upgrade pip
RUN python -m pip install onnxruntime-gpu==1.9.0 numpy==1.21.2 h5py numexpr opencv-python==4.5.3.56 opencv-contrib-python==4.5.3.56 pyqt6==6.2.1 onnx==1.10.1 torch==1.8.1 torchvision==0.9.1
RUN python -m pip install onnxruntime-gpu==1.10.0 numpy==1.21.2 h5py numexpr opencv-python==4.5.3.56 opencv-contrib-python==4.5.3.56 pyqt6==6.2.1 onnx==1.10.1 torch==1.8.1 torchvision==0.9.1
WORKDIR /app/DeepFaceLive
COPY example.sh example.sh

4
build/windows/WindowsBuilder.py
View file

@ -481,9 +481,9 @@ 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.9.0')
builder.install_pip_package('onnxruntime-gpu==1.10.0')
elif backend == 'directml':
builder.install_pip_package('onnxruntime-directml==1.9.0')
builder.install_pip_package('onnxruntime-directml==1.10.0')
builder.install_ffmpeg_binaries()