update packages versions

This commit is contained in:
iperov 2022-05-03 12:41:52 +04:00
parent f5b6447cbb
commit 534b9bcc96
2 changed files with 4 additions and 4 deletions

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.11.0 numpy==1.21.5 h5py numexpr opencv-python==4.5.5.64 opencv-contrib-python==4.5.5.64 pyqt6==6.2.3 onnx==1.11.0 torch==1.8.1 torchvision==0.9.1
RUN python -m pip install onnxruntime-gpu==1.11.1 numpy==1.21.5 h5py numexpr opencv-python==4.5.5.64 opencv-contrib-python==4.5.5.64 pyqt6==6.3.0 onnx==1.11.0 torch==1.8.1 torchvision==0.9.1
WORKDIR /app/DeepFaceLive
COPY example.sh example.sh

View file

@ -476,14 +476,14 @@ def build_deepfacelive_windows(release_dir, cache_dir, python_ver='3.7.9', backe
builder.install_pip_package('numexpr')
builder.install_pip_package('opencv-python==4.5.5.64')
builder.install_pip_package('opencv-contrib-python==4.5.5.64')
builder.install_pip_package('pyqt6==6.2.3')
builder.install_pip_package('pyqt6==6.3.0')
builder.install_pip_package('onnx==1.11.0')
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.11.0')
builder.install_pip_package('onnxruntime-gpu==1.11.1')
elif backend == 'directml':
builder.install_pip_package('onnxruntime-directml==1.11.0')
builder.install_pip_package('onnxruntime-directml==1.11.1')
builder.install_ffmpeg_binaries()