mirror of
https://github.com/iperov/DeepFaceLive
synced 2025-07-15 09:33:45 -07:00
update modules versions
This commit is contained in:
parent
d66ff580d7
commit
b7103b276c
2 changed files with 11 additions and 10 deletions
|
@ -10,7 +10,7 @@ RUN ln -s /usr/bin/python3 /usr/bin/python
|
||||||
RUN git clone https://github.com/iperov/DeepFaceLive.git
|
RUN git clone https://github.com/iperov/DeepFaceLive.git
|
||||||
|
|
||||||
RUN python -m pip install --upgrade pip
|
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.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 python -m pip install onnxruntime-gpu==1.14.1 numpy==1.21.6 h5py numexpr protobuf==3.20.1 opencv-python==4.7.0.72 opencv-contrib-python==4.7.0.72 pyqt6==6.5.0 onnx==1.13.0 torch==1.13.1 torchvision==0.14.1
|
||||||
|
|
||||||
RUN apt install -y libnvidia-compute-$NV_VER
|
RUN apt install -y libnvidia-compute-$NV_VER
|
||||||
|
|
||||||
|
|
|
@ -1,6 +1,7 @@
|
||||||
import argparse
|
import argparse
|
||||||
import os
|
import os
|
||||||
import shutil
|
import shutil
|
||||||
|
import ssl
|
||||||
import subprocess
|
import subprocess
|
||||||
import time
|
import time
|
||||||
import urllib.request
|
import urllib.request
|
||||||
|
@ -81,7 +82,7 @@ class WindowsFolderBuilder:
|
||||||
f = None
|
f = None
|
||||||
while True:
|
while True:
|
||||||
try:
|
try:
|
||||||
url_request = urllib.request.urlopen(url)
|
url_request = urllib.request.urlopen(url, context=ssl._create_unverified_context())
|
||||||
url_size = int( url_request.getheader('content-length') )
|
url_size = int( url_request.getheader('content-length') )
|
||||||
|
|
||||||
if use_cached:
|
if use_cached:
|
||||||
|
@ -462,7 +463,7 @@ pause
|
||||||
""")
|
""")
|
||||||
|
|
||||||
|
|
||||||
def build_deepfacelive_windows(release_dir, cache_dir, python_ver='3.7.9', backend='cuda'):
|
def build_deepfacelive_windows(release_dir, cache_dir, python_ver='3.8.10', backend='cuda'):
|
||||||
|
|
||||||
builder = WindowsFolderBuilder(release_path=Path(release_dir),
|
builder = WindowsFolderBuilder(release_path=Path(release_dir),
|
||||||
cache_path=Path(cache_dir),
|
cache_path=Path(cache_dir),
|
||||||
|
@ -475,19 +476,19 @@ def build_deepfacelive_windows(release_dir, cache_dir, python_ver='3.7.9', backe
|
||||||
builder.install_pip_package('h5py')
|
builder.install_pip_package('h5py')
|
||||||
builder.install_pip_package('numexpr')
|
builder.install_pip_package('numexpr')
|
||||||
builder.install_pip_package('protobuf==3.20.1')
|
builder.install_pip_package('protobuf==3.20.1')
|
||||||
builder.install_pip_package('opencv-python==4.7.0.68')
|
builder.install_pip_package('opencv-python==4.7.0.72')
|
||||||
builder.install_pip_package('opencv-contrib-python==4.7.0.68')
|
builder.install_pip_package('opencv-contrib-python==4.7.0.72')
|
||||||
builder.install_pip_package('pyqt6==6.4.0')
|
builder.install_pip_package('pyqt6==6.5.0')
|
||||||
builder.install_pip_package('onnx==1.13.0')
|
builder.install_pip_package('onnx==1.13.0')
|
||||||
|
|
||||||
if backend == 'cuda':
|
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')
|
#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('torch==1.11.0+cu115 torchvision==0.12.0+cu115 -f https://download.pytorch.org/whl/torch_stable.html')
|
#builder.install_pip_package('torch==1.11.0+cu115 torchvision==0.12.0+cu115 -f https://download.pytorch.org/whl/torch_stable.html')
|
||||||
builder.install_pip_package('torch==1.12.1+cu116 torchvision==0.13.1+cu116 -f https://download.pytorch.org/whl/torch_stable.html')
|
builder.install_pip_package('torch==1.13.1+cu117 torchvision==0.14.1+cu117 -f https://download.pytorch.org/whl/torch_stable.html')
|
||||||
|
|
||||||
builder.install_pip_package('onnxruntime-gpu==1.13.1')
|
builder.install_pip_package('onnxruntime-gpu==1.14.1')
|
||||||
elif backend == 'directml':
|
elif backend == 'directml':
|
||||||
builder.install_pip_package('onnxruntime-directml==1.13.1')
|
builder.install_pip_package('onnxruntime-directml==1.14.1')
|
||||||
|
|
||||||
builder.install_ffmpeg_binaries()
|
builder.install_ffmpeg_binaries()
|
||||||
|
|
||||||
|
@ -562,7 +563,7 @@ if __name__ == '__main__':
|
||||||
p.add_argument('--build-type', required=True, choices=['dfl-windows'])
|
p.add_argument('--build-type', required=True, choices=['dfl-windows'])
|
||||||
p.add_argument('--release-dir', action=fixPathAction, default=None)
|
p.add_argument('--release-dir', action=fixPathAction, default=None)
|
||||||
p.add_argument('--cache-dir', action=fixPathAction, default=None)
|
p.add_argument('--cache-dir', action=fixPathAction, default=None)
|
||||||
p.add_argument('--python-ver', default="3.7.9")
|
p.add_argument('--python-ver', default="3.8.10")
|
||||||
p.add_argument('--backend', choices=['cuda', 'directml'], default='cuda')
|
p.add_argument('--backend', choices=['cuda', 'directml'], default='cuda')
|
||||||
|
|
||||||
args = p.parse_args()
|
args = p.parse_args()
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue