diff --git a/build/linux/Dockerfile b/build/linux/Dockerfile index 5a707d1..6b18607 100644 --- a/build/linux/Dockerfile +++ b/build/linux/Dockerfile @@ -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 diff --git a/build/windows/WindowsBuilder.py b/build/windows/WindowsBuilder.py index 1f94377..570dd8d 100644 --- a/build/windows/WindowsBuilder.py +++ b/build/windows/WindowsBuilder.py @@ -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()