mirror of
https://github.com/iperov/DeepFaceLive
synced 2025-08-20 13:33:25 -07:00
removing CuPy. FaceMerger now works with any OpenCL1.2-compatible GPU.
This commit is contained in:
parent
c2ba2bab9d
commit
4fe2da23c0
6 changed files with 134 additions and 292 deletions
|
@ -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.8.1 cupy-cuda113 numpy==1.21.2 scipy==1.5.4 numexpr opencv-python==4.5.3.56 opencv-contrib-python==4.5.3.56 pyqt6==6.1.1 onnx==1.10.1 torch==1.8.1 torchvision==0.9.1
|
||||
RUN python -m pip install onnxruntime-gpu==1.8.1 numpy==1.21.2 numexpr opencv-python==4.5.3.56 opencv-contrib-python==4.5.3.56 pyqt6==6.1.1 onnx==1.10.1 torch==1.8.1 torchvision==0.9.1
|
||||
|
||||
WORKDIR /app/DeepFaceLive
|
||||
COPY example.sh example.sh
|
||||
|
|
|
@ -472,7 +472,6 @@ def build_deepfacelive_windows(release_dir, cache_dir, python_ver='3.7.9', backe
|
|||
# PIP INSTALLATIONS
|
||||
|
||||
builder.install_pip_package('numpy==1.21.2')
|
||||
builder.install_pip_package('scipy==1.5.4')
|
||||
builder.install_pip_package('numexpr')
|
||||
builder.install_pip_package('opencv-python==4.5.3.56')
|
||||
builder.install_pip_package('opencv-contrib-python==4.5.3.56')
|
||||
|
@ -482,7 +481,6 @@ 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('cupy-cuda111===9.0.0')
|
||||
elif backend == 'directml':
|
||||
if python_ver[:3] == '3.7':
|
||||
builder.install_pip_package('https://github.com/iperov/DeepFaceLive/releases/download/ort-dml/onnxruntime_directml-1.8.2-cp37-cp37m-win_amd64.whl')
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue