setup environment

This commit is contained in:
zpeng11 committed 2024-12-31 12:08:27 -05:00
1 parent 85b5f3182f
commit 1db66c4d2e
5 files changed
+19 -5

No files matched your search

@@ -37,4 +37,6 @@ echo yes|python -m pip install nvidia-cudnn-cu12
echo yes|pip install tensorrt_cu12_libs==10.6.0 tensorrt_cu12_bindings==10.6.0 tensorrt==10.6.0 --extra-index-url https://pypi.nvidia.com
call python -m pip install -r requirements.txt --no-warn-script-location
call python ezvtb_rt_interface.py
pause
+1 -1
View File
@@ -6,4 +6,4 @@ call conda env list
call conda activate ezvtb_rt_venv
call conda env list
python launcher.py
pythonw launcher.py
+8 -2
View File
@@ -1,3 +1,9 @@
@echo off
bin\python.exe launcher.py
pause
cd /D "%~dp0"
IF EXIST %~dp0envs\miniconda3\Scripts SET PATH=%~dp0envs\miniconda3\Scripts;%PATH%
call activate
call conda env list
call conda activate ezvtb_rt_venv
call conda env list
python launcher.py
+7 -1
View File
@@ -137,4 +137,10 @@ def get_core(
core = CoreORT(tha_model_dir, rife_path = rife_model_path if len(rife_model_path) > 0 else None, sr_path = sr_model_path if len(sr_model_path) > 0 else None, cacher=cacher, device_id=device_id)
return core
if __name__ == '__main__':
from ezvtb_rt.trt_utils import check_build_all_models
try:
check_build_all_models()
except:
pass