diff --git a/build/linux/Dockerfile b/build/linux/Dockerfile index 5cb56f4..dc9ff50 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.11.1 numpy==1.21.6 h5py numexpr protobuf==3.20.1 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 +RUN python -m pip install onnxruntime-gpu==1.11.1 numpy==1.21.6 h5py numexpr protobuf==3.20.1 opencv-python==4.5.5.64 opencv-contrib-python==4.5.5.64 pyqt6==6.3.0 onnx==1.11.0 torch==1.10.0 torchvision==0.11.1 WORKDIR /app/DeepFaceLive COPY example.sh example.sh diff --git a/build/windows/WindowsBuilder.py b/build/windows/WindowsBuilder.py index 98cb9fb..7a00286 100644 --- a/build/windows/WindowsBuilder.py +++ b/build/windows/WindowsBuilder.py @@ -481,7 +481,7 @@ def build_deepfacelive_windows(release_dir, cache_dir, python_ver='3.7.9', backe 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('torch==1.10.0+cu113 torchvision==0.11.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html') builder.install_pip_package('onnxruntime-gpu==1.11.1') elif backend == 'directml': builder.install_pip_package('onnxruntime-directml==1.11.1')