update modules versions

This commit is contained in:
iperov 2023-01-21 12:48:25 +04:00
parent 1f1b7bf376
commit e6787518d3
2 changed files with 4 additions and 4 deletions

View file

@ -10,7 +10,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.12.1 numpy==1.21.6 h5py numexpr protobuf==3.20.1 opencv-python==4.6.0.66 opencv-contrib-python==4.6.0.66 pyqt6==6.4.0 onnx==1.12.0 torch==1.10.0 torchvision==0.11.1
RUN python -m pip install onnxruntime-gpu==1.12.1 numpy==1.21.6 h5py numexpr protobuf==3.20.1 opencv-python==4.7.0.68 opencv-contrib-python==4.7.0.68 pyqt6==6.4.0 onnx==1.13.0 torch==1.10.0 torchvision==0.11.1
RUN apt install -y libnvidia-compute-$NV_VER

View file

@ -475,10 +475,10 @@ def build_deepfacelive_windows(release_dir, cache_dir, python_ver='3.7.9', backe
builder.install_pip_package('h5py')
builder.install_pip_package('numexpr')
builder.install_pip_package('protobuf==3.20.1')
builder.install_pip_package('opencv-python==4.6.0.66')
builder.install_pip_package('opencv-contrib-python==4.6.0.66')
builder.install_pip_package('opencv-python==4.7.0.68')
builder.install_pip_package('opencv-contrib-python==4.7.0.68')
builder.install_pip_package('pyqt6==6.4.0')
builder.install_pip_package('onnx==1.12.0')
builder.install_pip_package('onnx==1.13.0')
if backend == 'cuda':
#builder.install_pip_package('torch==1.10.0+cu113 torchvision==0.11.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html')