mirror of
https://github.com/hacksider/Deep-Live-Cam.git
synced 2026-06-02 10:41:44 +02:00
Kenneth Estanislao
41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import sys
|
|
|
|
# Add the project root to PATH so bundled ffmpeg/ffprobe are found
|
|
project_root = os.path.dirname(os.path.abspath(__file__))
|
|
os.environ["PATH"] = project_root + os.pathsep + os.environ.get("PATH", "")
|
|
|
|
# On Windows, register NVIDIA CUDA DLL directories so onnxruntime-gpu can
|
|
# find cuDNN/cublas. Python 3.8+ ignores PATH for extension-module native deps —
|
|
# os.add_dll_directory() is required. Also keep PATH for child processes/ffmpeg.
|
|
if sys.platform == "win32":
|
|
_site_packages = os.path.join(sys.prefix, "Lib", "site-packages")
|
|
_venv_site_packages = os.path.join(project_root, "venv", "Lib", "site-packages")
|
|
for _sp in (_site_packages, _venv_site_packages):
|
|
_candidate_dirs = []
|
|
_torch_lib = os.path.join(_sp, "torch", "lib")
|
|
if os.path.isdir(_torch_lib):
|
|
_candidate_dirs.append(_torch_lib)
|
|
_nvidia_dir = os.path.join(_sp, "nvidia")
|
|
if os.path.isdir(_nvidia_dir):
|
|
for _pkg in os.listdir(_nvidia_dir):
|
|
_bin_dir = os.path.join(_nvidia_dir, _pkg, "bin")
|
|
if os.path.isdir(_bin_dir):
|
|
_candidate_dirs.append(_bin_dir)
|
|
for _d in _candidate_dirs:
|
|
os.environ["PATH"] = _d + os.pathsep + os.environ["PATH"]
|
|
try:
|
|
os.add_dll_directory(_d)
|
|
except (OSError, AttributeError):
|
|
pass
|
|
|
|
from modules import platform_info
|
|
platform_info.print_banner()
|
|
|
|
from modules import core
|
|
|
|
if __name__ == '__main__':
|
|
core.run()
|