diff --git a/00.检查安装CudaToolkits.bat b/00.检查安装CudaToolkits.bat new file mode 100644 index 0000000..51e6973 --- /dev/null +++ b/00.检查安装CudaToolkits.bat @@ -0,0 +1,12 @@ +@echo off + +WHERE nvidia-smi +IF %ERRORLEVEL% NEQ 0 ( + echo 'CudaTookit No Ready!!!! Please install manually' +) else ( + echo 'CudaTookit Ready, go ahead next step' +) +echo. +echo. +echo Make sure everything is OK before you run 01A or 01B. +pause \ No newline at end of file diff --git a/00.检查并补齐必需文件.bat b/00.检查并补齐必需文件.bat deleted file mode 100644 index ea5eb51..0000000 --- a/00.检查并补齐必需文件.bat +++ /dev/null @@ -1,11 +0,0 @@ -@echo off -echo 1. Pretrained data -if exist data\models\standard_float\editor.pt ( -echo OK! Pretrained data found. -) else ( -echo Pretrained data not found, use link 00B and extract it to data/models.. -) -echo. -echo. -echo Make sure everything is OK before you run 01A or 01B. -pause \ No newline at end of file diff --git a/00A.英伟达CudaToolkits.url b/00A.英伟达CudaToolkits.url new file mode 100644 index 0000000..4cc81d2 --- /dev/null +++ b/00A.英伟达CudaToolkits.url @@ -0,0 +1,5 @@ +[{000214A0-0000-0000-C000-000000000046}] +Prop3=19,11 +[InternetShortcut] +IDList= +URL=https://developer.nvidia.com/cuda-downloads diff --git a/00B.Spout2插件.url b/00B.Spout2插件.url new file mode 100644 index 0000000..af164f4 --- /dev/null +++ b/00B.Spout2插件.url @@ -0,0 +1,5 @@ +[{000214A0-0000-0000-C000-000000000046}] +Prop3=19,11 +[InternetShortcut] +IDList= +URL=https://github.com/Off-World-Live/obs-spout2-plugin/releases diff --git a/00B.预训练数据下载（需要科学上网）.url b/00B.预训练数据下载（需要科学上网）.url deleted file mode 100644 index acb8478..0000000 --- a/00B.预训练数据下载（需要科学上网）.url +++ /dev/null @@ -1,6 +0,0 @@ -[{000214A0-0000-0000-C000-000000000046}] -Prop3=19,11 -[InternetShortcut] -IDList= -URL=https://www.dropbox.com/s/y7b8jl4n2euv8xe/talking-head-anime-3-models.zip?dl=0 -HotKey=0 diff --git a/01A.构建运行环境（默认源）.bat b/01A.构建运行环境（默认源）.bat index 91caddb..0837ecb 100644 --- a/01A.构建运行环境（默认源）.bat +++ b/01A.构建运行环境（默认源）.bat @@ -29,7 +29,6 @@ call conda activate ezvtb_rt_venv call conda env list call conda install -y conda-forge::pycuda -call conda install -y pytorch-cpu call python -m pip install --upgrade pip wheel echo yes|python -m pip install nvidia-cudnn-cu12 @@ -38,5 +37,4 @@ echo yes|pip install tensorrt_cu12_libs==10.6.0 tensorrt_cu12_bindings==10.6.0 t call python -m pip install -r requirements.txt --no-warn-script-location -call python ezvtb_rt_interface.py pause \ No newline at end of file diff --git a/01B.构建运行环境（国内源）.bat b/01B.构建运行环境（国内源）.bat index 9bba421..7a0eedc 100644 --- a/01B.构建运行环境（国内源）.bat +++ b/01B.构建运行环境（国内源）.bat @@ -29,7 +29,6 @@ call conda activate ezvtb_rt_venv call conda env list call conda install -y pycuda -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ -call conda install -y pytorch-cpu -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ call python -m pip install --upgrade pip wheel -i https://mirrors.aliyun.com/pypi/simple/ echo yes|python -m pip install nvidia-cudnn-cu12 -i https://mirrors.aliyun.com/pypi/simple/ @@ -38,5 +37,4 @@ echo yes|pip install tensorrt_cu12_libs==10.6.0 tensorrt_cu12_bindings==10.6.0 t call python -m pip install -r requirements.txt --no-warn-script-location -i https://mirrors.aliyun.com/pypi/simple/ -call python ezvtb_rt_interface.py pause \ No newline at end of file diff --git a/01X.清理PIP下载缓存（重试也解决不了问题时使用）.bat b/01X.清理PIP下载缓存（重试也解决不了问题时使用）.bat deleted file mode 100644 index 953be69..0000000 --- a/01X.清理PIP下载缓存（重试也解决不了问题时使用）.bat +++ /dev/null @@ -1,2 +0,0 @@ -cd bin -python.exe -m pip cache purge \ No newline at end of file diff --git a/02A.启动器.bat b/02A.启动器.bat index aa31f5e..c79afba 100644 --- a/02A.启动器.bat +++ b/02A.启动器.bat @@ -6,4 +6,6 @@ call conda env list call conda activate ezvtb_rt_venv call conda env list -pythonw launcher.py \ No newline at end of file +IF not EXIST %~dp0launcher.json call python ezvtb_rt_interface.py + +pythonw launcher2.py \ No newline at end of file diff --git a/02B.启动器（调试输出）.bat b/02B.启动器（调试输出）.bat index dded99e..16cd3c4 100644 --- a/02B.启动器（调试输出）.bat +++ b/02B.启动器（调试输出）.bat @@ -1,4 +1,4 @@ -@echo off +@echo on cd /D "%~dp0" IF EXIST %~dp0envs\miniconda3\Scripts SET PATH=%~dp0envs\miniconda3\Scripts;%PATH% call activate @@ -6,4 +6,8 @@ call conda env list call conda activate ezvtb_rt_venv call conda env list -python launcher.py \ No newline at end of file +IF not EXIST %~dp0launcher.json call python ezvtb_rt_interface.py + +python launcher2.py + +pause \ No newline at end of file diff --git a/README.md b/README.md index e84d2f3..5ea1cb4 100644 --- a/README.md +++ b/README.md @@ -12,17 +12,19 @@ Fork自 https://github.com/GunwooHan/EasyVtuber [视频介绍和安装说明](https://www.bilibili.com/video/BV1uu411r7DR) Updates: -使用 https://github.com/zpeng11/ezvtuber-rt/tree/main 项目转换的ONNX模型，实现TensorRT加速以及非N卡支持，在半精度和全精度下均有加速（具体取决于显卡） -为A卡和I卡提供DirectML支持，人人都能玩。 -引入RIFE模型进行插帧，极限帧数提升达到50%-100%，配合锁帧可以有效降低使用占用。 -使用 waifu2x 和 real-esrgan 项目带来的输出超分辨率，对比anime4k效果提升显著（也使用更多gpu占用） +* 使用 https://github.com/zpeng11/ezvtuber-rt/tree/main 项目转换的ONNX模型，实现TensorRT加速以及非N卡支持，在半精度和全精度下均有加速（具体取决于显卡） +* 为A卡和I卡提供DirectML支持，人人都能玩。 +* 引入[RIFE](https://github.com/hzwer/ECCV2022-RIFE)模型进行插帧，极限帧数提升达到50%-100%，配合锁帧可以有效降低使用占用。 +* 使用 [TurboJPEG](https://github.com/Dobatymo/turbojpeg-python) 获得成倍缓存命中率提升，在长时间使用上显卡减负效果相当出色。 +* 使用 [waifu2x](https://github.com/nagadomi/waifu2x) 和 [real-esrgan]() 项目带来的输出超分辨率，对比anime4k效果提升显著（也使用更多gpu占用） +* 更漂亮的 WxPython 新启动器界面，带中文介绍更加简单易用 ## Requirements ### 硬件 - 支持FaceID的iPhone（使用ifacialmocap软件，需购买，需要稳定的WIFI连接）或网络摄像头（使用OpenCV） -- 任意5年内的游戏级显卡 +- 任意5年内的游戏级显卡，N卡/A卡/I卡均可使用 ### 软件 - 本方案在Windows 10上测试可用 @@ -31,68 +33,132 @@ Updates: - Photoshop或其他图片处理软件 - 科学上网方案，看懂英文网站和报错的能力 -## Installation(Release包版本) -Release包版本带有本项目源码，ezvtuber-rt项目源码，和静态化模型。并且可以自动构建python环境。 -构建运行环境依赖anaconda，若本机未安装anaconda将自动下载miniconda并在`envs`文件夹内展开。 - +## 整合包版本 +请使用合适方式下载整合包并自行解压： +* [夸克网盘](https://pan.quark.cn/s/b61ad5315f59) +* 磁力链接 magnet:?xt=urn:btih:31ea4e070603b02a2ffd38418116b08b74551722 +* [谷歌网盘](https://drive.google.com/drive/folders/1cYj18EfVQ2Cl348_rkCu_fgaasHTI_io?usp=drive_link) ### 下载并安装CUDAToolkit -前往 [英伟达官网](https://developer.nvidia.com/cuda-downloads) 下载并安装`CUDATookit 12`及以上版本。 +前往 [英伟达官网](https://developer.nvidia.com/cuda-downloads) 下载并安装`CUDATookit 12`及以上版本。 +若安装成功并重启可开启命令行工具验证出现类似如下结果： +``` +C:\Users\Eleven>nvidia-smi +Thu Jan 9 22:02:29 2025 ++-----------------------------------------------------------------------------------------+ +| NVIDIA-SMI 556.12 Driver Version: 556.12 CUDA Version: 12.5 | +|-----------------------------------------+------------------------+----------------------+ +| GPU Name Driver-Model | Bus-Id Disp.A | Volatile Uncorr. ECC | +| Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. | +| | | MIG M. | +|=========================================+========================+======================| +| 0 NVIDIA GeForce GTX 1660 ... WDDM | 00000000:01:00.0 On | N/A | +| 32% 34C P8 10W / 125W | 569MiB / 6144MiB | 1% Default | +| | | N/A | ++-----------------------------------------+------------------------+----------------------+ +........ +``` +或可以双击 `00.检查安装CudaToolkits.bat` 来验证 -### 下载ZIP并解压 -点击[`Download ZIP`](../../archive/master.zip) 下载并解压，源码和模型约3.5G。若在文件夹内展开miniconda和所有运行环境总计需要约20GB - -### 构建运行环境 -双击运行适合你的地域的`01A.构建运行环境（默认源）.bat`或者`01B.构建运行环境（国内源）.bat` -这个脚本会使用anaconda安装所有需要的依赖 -两个脚本可以互相替代，并且支持从中断的位置继续 -如果出现网络相关报错，直接关掉控制台，调整网络，重新运行即可 -![step01success](assets/01Success.png) -完全安装完成后再次运行脚本的输出如图所示。一般来说安装全程没有红字就是成功结束。 -若是英伟达用户将会需要较长时间对模型进行TensorRT编译（>30min） - -### 使用启动器测试结果 -运行`02B启动器（调试输出）.bat` -直接点击界面底部的`Save & Launch` -如果看到了弹出的opencv输出窗体，则安装成功完成 -![img.png](assets/02success.png) - -### 配置输入输出设备 -在成功进行Debug输出之后，请移步之后的输入输出设备一节进行进一步配置以输出到OBS。 +### 安装Spout2 OBS插件(可选) +安装整合包附带的 `OBS_Spout2_Plugin_Install_v1.8.exe`， 安装包来自 https://github.com/Off-World-Live/obs-spout2-plugin 项目，可以提供透明通道给OBS。 +### 启动项目 +双击`02A.启动器.bat` 或 `02B.启动器（调试输出）.bat` 启动器，首次启动将会花约5分钟安装pip资源， 若是英伟达用户将会开始TensorRT构建，额外需要>20min时间(取决于显卡)，结束后可以大幅提升运行速度，构建的开头和结束输出如下。 +``` +Start testing if TensorRT works on this machine + 0%| | 0/38 [00:00conda --version +conda 24.11.3 +``` +以上方式可以保证conda环境被你选定的Ananconda统一管理。若没有此需求可以忽略直接运行下一步，脚本会自动安装miniconda。 ### 克隆项目和子项目 ``` git clone https://github.com/zpeng11/EasyVtuber.git +cd EasyVtuber git submodule init git submodule update --recursive --remote -cd EasyVtuber ``` -### 使用脚本安装环境并下载模型 +### 安装环境 双击运行`01A.构建运行环境（默认源）.bat`或者`01B.构建运行环境（国内源）.bat` -这两个脚本会构建名为`ezvtb_rt_venv`的conda环境，并从Github Release下载模型 +这两个脚本会构建名为`ezvtb_rt_venv`的conda环境 +或可以用如下命令手动创建： +``` +conda create -y -n ezvtb_rt_venv python=3.10 +conda activate ezvtb_rt_venv +conda install -y conda-forge::pycuda +python -m pip install --upgrade pip wheel +python -m pip install nvidia-cudnn-cu12 +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 +python -m pip install -r requirements.txt --no-warn-script-location +``` + +### 下载模型 +前往模型 release 地址 https://github.com/zpeng11/ezvtuber-rt/releases/tag/0.0.1 +下载 https://github.com/zpeng11/ezvtuber-rt/releases/download/0.0.1/20241220.zip 并解压到`data/models`文件夹下 ### Pycharm 配置 安装完成后，在Pycharm内打开本项目，右下角解释器菜单点开，`Add Interpreter...`->`Conda Environment`->`Existing environment` 选好自己电脑上的`conda.exe`和刚才创建好的`ezvtb_rt_venv`环境内的`python.exe` 点击OK，依赖全亮即可 +### 检查环境模型，并构建TensorRT +在Conda环境中执行以下命令 +`python ezvtb_rt_interface.py` + ### 运行启动器 在Conda环境中执行以下命令 `python launcher.py` +### 错误排查 +当运行TensorRT构建或启动器出现错误时，请参考如下可能性： +1. 环境安装有错误（一般运行时缺少库99%的原因都是这个，pip下载并不稳定，各种网络问题都可能导致安装失败），检查源，手动删除`envs`文件夹再试, 对照 [此log](assets/complete_building_log.txt) 为成功案例排查原因。 实在不懂排查请使用整合包。 +2. nvcc编译器没找到，即CudaToolkit安装有误，双击 `00.检查安装CudaToolkits.bat` 来验证 +3. 英伟达显卡但计算架构低于7.5（10系以及之前）不支持TensorRT ## 输入输出设备 +#### SPOUT2 OBS插件输出 +目前推荐这个方案，使用透明通道效果最好也最方便。使用前请移步 https://github.com/Off-World-Live/obs-spout2-plugin/releases 安装obs插件，并在obs中选择spout输入设备。 #### OBS Virtual Camera -目前更推荐这个方案，UnityCapture存在未查明的性能瓶颈 +UnityCapture存在未查明的性能瓶颈 如果你选择自己进行抠像你可以直接输出到obs，如果你需要RGBA支持则需要额外使用一个Shader 下载并安装StreamFX https://github.com/Xaymar/obs-StreamFX 下载Shader（感谢树根的协助） https://github.com/shugen002/shader/blob/master/merge%20alpha2.hlsl @@ -131,7 +197,7 @@ https://github.com/emilianavt/OpenSeeFace/releases ## Run -完全体运行命令`python main.py --output_webcam unitycapture --ifm 192.168.31.182:49983 --character test1L2 --extend_movement 1 --output_size 512x512` +`main.py` 的完全体运行命令请参考 `02B.启动器（调试输出）.bat` 的输出以及 `args.py`。 以下为部分基础参数 参数名 | 值类型 | 说明 :---: | :---: | :---: diff --git a/args.py b/args.py index e92bb08..46dd57c 100644 --- a/args.py +++ b/args.py @@ -75,6 +75,3 @@ args.model_output_size = 1024 if args.use_sr else 512 if args.use_sr and args.use_interpolation: args.use_interpolation = False -# if args.use_sr and args.sr_x4: -# print('debug') -# args.cacher_quality = 100 diff --git a/assets/00Success.png b/assets/00Success.png deleted file mode 100644 index 8f33bf1..0000000 Binary files a/assets/00Success.png and /dev/null differ diff --git a/assets/01Success.png b/assets/01Success.png deleted file mode 100644 index 23b86dc..0000000 Binary files a/assets/01Success.png and /dev/null differ diff --git a/assets/02success.png b/assets/02success.png index c73dc0f..3b0d1bf 100644 Binary files a/assets/02success.png and b/assets/02success.png differ diff --git a/assets/complete_building_log.txt b/assets/complete_building_log.txt new file mode 100644 index 0000000..dcc6465 --- /dev/null +++ b/assets/complete_building_log.txt @@ -0,0 +1,548 @@ +C:\Users\Eleven\Desktop\EasyVtuber-main>cd /D "C:\Users\Eleven\Desktop\EasyVtuber-main\" + +C:\Users\Eleven\Desktop\EasyVtuber-main>WHERE conda +INFO: Could not find files for the given pattern(s). + +C:\Users\Eleven\Desktop\EasyVtuber-main>IF 1 NEQ 0 ( +if not exist "envs" mkdir envs + if not exist "envs\miniconda3" mkdir envs\miniconda3 + IF not EXIST C:\Users\Eleven\Desktop\EasyVtuber-main\envs\miniconda3\Scripts ( + + mkdir C:\Users\Eleven\Desktop\EasyVtuber-main\envs\miniconda3 + echo "Downloading miniconda..." + powershell -Command "(New-Object Net.WebClient).DownloadFile('https://repo.anaconda.com/miniconda/Miniconda3-py310_24.9.2-0-Windows-x86_64.exe', '.\envs\miniconda3.exe')" + echo "Installling minconda..." + start /wait "" C:\Users\Eleven\Desktop\EasyVtuber-main\envs\miniconda3.exe /S /AddToPath=0 /RegisterPython=0 /InstallationType=JustMe /D=C:\Users\Eleven\Desktop\EasyVtuber-main\envs\miniconda3 + echo "Successfully install minconda" + del envs\miniconda3.exe +) +) +"Downloading miniconda..." +"Installling minconda..." +"Successfully install minconda" + +C:\Users\Eleven\Desktop\EasyVtuber-main>IF EXIST C:\Users\Eleven\Desktop\EasyVtuber-main\envs\miniconda3\Scripts SET PATH=C:\Users\Eleven\Desktop\EasyVtuber-main\envs\miniconda3\Scripts;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6\bin;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6\libnvvp;;C:\Windows\system32;C:\Windows;C:\Windows\System32\Wbem;C:\Windows\System32\WindowsPowerShell\v1.0\;C:\Windows\System32\OpenSSH\;C:\Program Files (x86)\NVIDIA Corporation\PhysX\Common;C:\Program Files\Git\cmd;C:\Program Files\Tailscale\;C:\Program Files\NVIDIA Corporation\Nsight Compute 2024.3.2\;C:\Program Files\NVIDIA Corporation\NVIDIA NvDLISR;C:\Program Files\dotnet\;C:\Users\Eleven\AppData\Local\Programs\Python\Python312\Scripts\;C:\Users\Eleven\AppData\Local\Programs\Python\Python312\;C:\Users\Eleven\.cargo\bin;C:\Users\Eleven\AppData\Local\Microsoft\WindowsApps;C:\Users\Eleven\AppData\Local\Programs\Microsoft VS Code\bin;C:\Program Files\Git\bin;D:\ffmpeg\bin; + +C:\Users\Eleven\Desktop\EasyVtuber-main>call activate +# conda environments: +# + C:\EasyVtuber\envs\miniconda3 + C:\EasyVtuber\envs\miniconda3\envs\ezvtb_rt_venv +base * C:\Users\Eleven\Desktop\EasyVtuber-main\envs\miniconda3 + +Retrieving notices: ...working... done +Channels: + - defaults +Platform: win-64 +Collecting package metadata (repodata.json): done +Solving environment: done + +## Package Plan ## + + environment location: C:\Users\Eleven\Desktop\EasyVtuber-main\envs\miniconda3 + + +The following packages will be downloaded: + + package | build + ---------------------------|----------------- + anaconda-anon-usage-0.5.0 |py310hfc23b7f_100 28 KB + brotli-python-1.0.9 | py310h5da7b33_9 346 KB + ca-certificates-2024.11.26 | haa95532_0 132 KB + certifi-2024.12.14 | py310haa95532_0 162 KB + cffi-1.17.1 | py310h827c3e9_1 254 KB + conda-24.11.2 | py310haa95532_0 948 KB + conda-libmamba-solver-24.11.1| pyhd3eb1b0_0 39 KB + conda-package-handling-2.4.0| py310haa95532_0 297 KB + conda-package-streaming-0.11.0| py310haa95532_0 31 KB + cpp-expected-1.1.0 | h214f63a_0 131 KB + cryptography-43.0.3 | py310hbd6ee87_1 1.3 MB + libarchive-3.7.7 | h9243413_0 1.8 MB + libcurl-8.11.1 | haff574d_0 382 KB + libmamba-2.0.5 | hcd6fe79_1 4.7 MB + libmambapy-2.0.5 | py310h214f63a_1 459 KB + libsolv-0.7.30 | hf2fb9eb_1 473 KB + libssh2-1.11.1 | h2addb87_0 312 KB + libxml2-2.13.5 | h24da03e_0 2.9 MB + menuinst-2.2.0 | py310h5da7b33_0 200 KB + nlohmann_json-3.11.2 | h6c2663c_0 126 KB + packaging-24.2 | py310haa95532_0 175 KB + pluggy-1.5.0 | py310haa95532_0 39 KB + requests-2.32.3 | py310haa95532_1 101 KB + simdjson-3.10.1 | h214f63a_0 275 KB + spdlog-1.11.0 | h59b6b97_0 195 KB + vc-14.40 | haa95532_2 10 KB + vs2015_runtime-14.42.34433 | h9531ae6_2 1.2 MB + zstandard-0.23.0 | py310h4fc1ca9_1 342 KB + ------------------------------------------------------------ + Total: 17.3 MB + +The following NEW packages will be INSTALLED: + + cpp-expected pkgs/main/win-64::cpp-expected-1.1.0-h214f63a_0 + nlohmann_json pkgs/main/win-64::nlohmann_json-3.11.2-h6c2663c_0 + simdjson pkgs/main/win-64::simdjson-3.10.1-h214f63a_0 + spdlog pkgs/main/win-64::spdlog-1.11.0-h59b6b97_0 + +The following packages will be UPDATED: + + anaconda-anon-usa~ 0.4.4-py310hfc23b7f_100 --> 0.5.0-py310hfc23b7f_100 + brotli-python 1.0.9-py310hd77b12b_8 --> 1.0.9-py310h5da7b33_9 + ca-certificates 2024.9.24-haa95532_0 --> 2024.11.26-haa95532_0 + certifi 2024.8.30-py310haa95532_0 --> 2024.12.14-py310haa95532_0 + cffi 1.17.1-py310h827c3e9_0 --> 1.17.1-py310h827c3e9_1 + conda 24.9.2-py310haa95532_0 --> 24.11.2-py310haa95532_0 + conda-libmamba-so~ 24.9.0-pyhd3eb1b0_0 --> 24.11.1-pyhd3eb1b0_0 + conda-package-han~ 2.3.0-py310haa95532_0 --> 2.4.0-py310haa95532_0 + conda-package-str~ 0.10.0-py310haa95532_0 --> 0.11.0-py310haa95532_0 + cryptography 43.0.0-py310h89fc84f_0 --> 43.0.3-py310hbd6ee87_1 + libarchive 3.7.4-h9243413_0 --> 3.7.7-h9243413_0 + libcurl 8.9.1-h0416ee5_0 --> 8.11.1-haff574d_0 + libmamba 1.5.8-h99b1521_3 --> 2.0.5-hcd6fe79_1 + libmambapy 1.5.8-py310h77c03ed_3 --> 2.0.5-py310h214f63a_1 + libsolv 0.7.24-h23ce68f_1 --> 0.7.30-hf2fb9eb_1 + libssh2 1.11.0-h291bd65_0 --> 1.11.1-h2addb87_0 + libxml2 2.13.1-h24da03e_2 --> 2.13.5-h24da03e_0 + menuinst 2.1.2-py310h5da7b33_0 --> 2.2.0-py310h5da7b33_0 + packaging 24.1-py310haa95532_0 --> 24.2-py310haa95532_0 + pluggy 1.0.0-py310haa95532_1 --> 1.5.0-py310haa95532_0 + requests 2.32.3-py310haa95532_0 --> 2.32.3-py310haa95532_1 + vc 14.40-h2eaa2aa_1 --> 14.40-haa95532_2 + vs2015_runtime 14.40.33807-h98bb1dd_1 --> 14.42.34433-h9531ae6_2 + zstandard 0.23.0-py310h4fc1ca9_0 --> 0.23.0-py310h4fc1ca9_1 + + + +Downloading and Extracting Packages: + +Preparing transaction: done +Verifying transaction: done +Executing transaction: done +Channels: + - defaults +Platform: win-64 +Collecting package metadata (repodata.json): done +Solving environment: done + +## Package Plan ## + + environment location: C:\Users\Eleven\Desktop\EasyVtuber-main\envs\miniconda3\envs\ezvtb_rt_venv + + added / updated specs: + - python=3.10 + + +The following packages will be downloaded: + + package | build + ---------------------------|----------------- + python-3.10.16 | h4607a30_1 16.3 MB + ------------------------------------------------------------ + Total: 16.3 MB + +The following NEW packages will be INSTALLED: + + bzip2 pkgs/main/win-64::bzip2-1.0.8-h2bbff1b_6 + ca-certificates pkgs/main/win-64::ca-certificates-2024.11.26-haa95532_0 + libffi pkgs/main/win-64::libffi-3.4.4-hd77b12b_1 + openssl pkgs/main/win-64::openssl-3.0.15-h827c3e9_0 + pip pkgs/main/win-64::pip-24.2-py310haa95532_0 + python pkgs/main/win-64::python-3.10.16-h4607a30_1 + setuptools pkgs/main/win-64::setuptools-75.1.0-py310haa95532_0 + sqlite pkgs/main/win-64::sqlite-3.45.3-h2bbff1b_0 + tk pkgs/main/win-64::tk-8.6.14-h0416ee5_0 + tzdata pkgs/main/noarch::tzdata-2024b-h04d1e81_0 + vc pkgs/main/win-64::vc-14.40-haa95532_2 + vs2015_runtime pkgs/main/win-64::vs2015_runtime-14.42.34433-h9531ae6_2 + wheel pkgs/main/win-64::wheel-0.44.0-py310haa95532_0 + xz pkgs/main/win-64::xz-5.4.6-h8cc25b3_1 + zlib pkgs/main/win-64::zlib-1.2.13-h8cc25b3_1 + + + +Downloading and Extracting Packages: + +Preparing transaction: done +Verifying transaction: done +Executing transaction: done +# +# To activate this environment, use +# +# $ conda activate ezvtb_rt_venv +# +# To deactivate an active environment, use +# +# $ conda deactivate + + +# conda environments: +# + C:\EasyVtuber\envs\miniconda3 + C:\EasyVtuber\envs\miniconda3\envs\ezvtb_rt_venv +base C:\Users\Eleven\Desktop\EasyVtuber-main\envs\miniconda3 +ezvtb_rt_venv * C:\Users\Eleven\Desktop\EasyVtuber-main\envs\miniconda3\envs\ezvtb_rt_venv + +Channels: + - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + - defaults +Platform: win-64 +Collecting package metadata (repodata.json): done +Solving environment: done + +## Package Plan ## + + environment location: C:\Users\Eleven\Desktop\EasyVtuber-main\envs\miniconda3\envs\ezvtb_rt_venv + + added / updated specs: + - pycuda + + +The following packages will be downloaded: + + package | build + ---------------------------|----------------- + appdirs-1.4.4 | pyhd8ed1ab_1 14 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + boost-1.78.0 | py310h220cb41_4 555 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + boost-cpp-1.78.0 | h5b4e17d_0 17.0 MB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + ca-certificates-2024.12.14 | h56e8100_0 154 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + cudatoolkit-11.8.0 | h09e9e62_11 638.9 MB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + importlib-metadata-8.5.0 | pyha770c72_1 28 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + intel-openmp-2024.2.1 | h57928b3_1083 1.8 MB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + libblas-3.9.0 | 26_win64_mkl 3.6 MB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + libcblas-3.9.0 | 26_win64_mkl 3.6 MB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + libhwloc-2.9.1 | h51c2c0f_0 2.4 MB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + libiconv-1.17 | h8ffe710_0 698 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + liblapack-3.9.0 | 26_win64_mkl 3.6 MB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + lz4-c-1.9.4 | hcfcfb64_0 131 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + mako-1.3.8 | pyhd8ed1ab_0 66 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + markupsafe-2.1.2 | py310h8d17308_0 25 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + mkl-2024.2.2 | h66d3029_15 98.3 MB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + numpy-1.24.3 | py310hd02465a_0 5.7 MB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + openssl-3.1.0 | hcfcfb64_3 7.1 MB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + platformdirs-4.3.6 | pyhd8ed1ab_1 20 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + pthreads-win32-2.9.1 | hfa6e2cd_3 141 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + pycuda-2022.2.2 | py310ha2c4f5d_0 378 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + python_abi-3.10 | 2_cp310 4 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + pytools-2024.1.6 | pyhd8ed1ab_0 72 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + tbb-2021.9.0 | h91493d7_0 151 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + typing_extensions-4.12.2 | pyha770c72_1 39 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + ucrt-10.0.22621.0 | h57928b3_1 547 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + zipp-3.21.0 | pyhd8ed1ab_1 21 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge + ------------------------------------------------------------ + Total: 784.9 MB + +The following NEW packages will be INSTALLED: + + appdirs anaconda/cloud/conda-forge/noarch::appdirs-1.4.4-pyhd8ed1ab_1 + boost anaconda/cloud/conda-forge/win-64::boost-1.78.0-py310h220cb41_4 + boost-cpp anaconda/cloud/conda-forge/win-64::boost-cpp-1.78.0-h5b4e17d_0 + cudatoolkit anaconda/cloud/conda-forge/win-64::cudatoolkit-11.8.0-h09e9e62_11 + importlib-metadata anaconda/cloud/conda-forge/noarch::importlib-metadata-8.5.0-pyha770c72_1 + intel-openmp anaconda/cloud/conda-forge/win-64::intel-openmp-2024.2.1-h57928b3_1083 + libblas anaconda/cloud/conda-forge/win-64::libblas-3.9.0-26_win64_mkl + libcblas anaconda/cloud/conda-forge/win-64::libcblas-3.9.0-26_win64_mkl + libhwloc anaconda/cloud/conda-forge/win-64::libhwloc-2.9.1-h51c2c0f_0 + libiconv anaconda/cloud/conda-forge/win-64::libiconv-1.17-h8ffe710_0 + liblapack anaconda/cloud/conda-forge/win-64::liblapack-3.9.0-26_win64_mkl + libxml2 pkgs/main/win-64::libxml2-2.13.5-h24da03e_0 + lz4-c anaconda/cloud/conda-forge/win-64::lz4-c-1.9.4-hcfcfb64_0 + mako anaconda/cloud/conda-forge/noarch::mako-1.3.8-pyhd8ed1ab_0 + markupsafe anaconda/cloud/conda-forge/win-64::markupsafe-2.1.2-py310h8d17308_0 + mkl anaconda/cloud/conda-forge/win-64::mkl-2024.2.2-h66d3029_15 + numpy anaconda/cloud/conda-forge/win-64::numpy-1.24.3-py310hd02465a_0 + platformdirs anaconda/cloud/conda-forge/noarch::platformdirs-4.3.6-pyhd8ed1ab_1 + pthreads-win32 anaconda/cloud/conda-forge/win-64::pthreads-win32-2.9.1-hfa6e2cd_3 + pycuda anaconda/cloud/conda-forge/win-64::pycuda-2022.2.2-py310ha2c4f5d_0 + python_abi anaconda/cloud/conda-forge/win-64::python_abi-3.10-2_cp310 + pytools anaconda/cloud/conda-forge/noarch::pytools-2024.1.6-pyhd8ed1ab_0 + tbb anaconda/cloud/conda-forge/win-64::tbb-2021.9.0-h91493d7_0 + typing_extensions anaconda/cloud/conda-forge/noarch::typing_extensions-4.12.2-pyha770c72_1 + ucrt anaconda/cloud/conda-forge/win-64::ucrt-10.0.22621.0-h57928b3_1 + zipp anaconda/cloud/conda-forge/noarch::zipp-3.21.0-pyhd8ed1ab_1 + zstd pkgs/main/win-64::zstd-1.5.6-h8880b57_0 + +The following packages will be UPDATED: + + ca-certificates pkgs/main::ca-certificates-2024.11.26~ --> anaconda/cloud/conda-forge::ca-certificates-2024.12.14-h56e8100_0 + openssl pkgs/main::openssl-3.0.15-h827c3e9_0 --> anaconda/cloud/conda-forge::openssl-3.1.0-hcfcfb64_3 + + + +Downloading and Extracting Packages: + +Preparing transaction: done +Verifying transaction: done +Executing transaction: | "By downloading and using the CUDA Toolkit conda packages, you accept the terms and conditions of the CUDA End User License Agreement (EULA): https://docs.nvidia.com/cuda/eula/index.html" + +done +Channels: + - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + - defaults +Platform: win-64 +Collecting package metadata (repodata.json): done +Solving environment: done + +## Package Plan ## + + environment location: C:\Users\Eleven\Desktop\EasyVtuber-main\envs\miniconda3\envs\ezvtb_rt_venv + + added / updated specs: + - pytorch-cpu + + +The following packages will be downloaded: + + package | build + ---------------------------|----------------- + blas-1.0 | mkl 6 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + filelock-3.13.1 | py310haa95532_0 22 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + fsspec-2024.6.1 | py310haa95532_0 286 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + gmpy2-2.1.2 | py310h7f96b67_0 160 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + intel-openmp-2023.1.0 | h59b6b97_46320 2.7 MB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + jinja2-3.1.4 | py310haa95532_1 281 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + libuv-1.48.0 | h827c3e9_0 322 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + mkl-2023.1.0 | h6b88ed4_46358 155.9 MB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + mkl-service-2.4.0 | py310h2bbff1b_1 44 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + mkl_fft-1.3.11 | py310h827c3e9_0 168 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + mkl_random-1.2.8 | py310hc64d2fc_0 257 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + mpc-1.1.0 | h7edee0f_1 260 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + mpfr-4.0.2 | h62dcd97_1 1.5 MB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + mpir-3.0.0 | hec2e145_1 1.3 MB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + mpmath-1.3.0 | py310haa95532_0 834 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + networkx-3.3 | py310haa95532_0 2.5 MB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + numpy-1.26.4 | py310h055cbcc_0 11 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + numpy-base-1.26.4 | py310h65a83cf_0 8.6 MB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + pytorch-2.3.1 |cpu_py310h0ce1571_0 111.2 MB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + pytorch-cpu-2.3.1 | haa95532_0 21 KB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + sympy-1.13.3 | py310haa95532_0 11.3 MB https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main + ------------------------------------------------------------ + Total: 297.6 MB + +The following NEW packages will be INSTALLED: + + blas anaconda/pkgs/main/win-64::blas-1.0-mkl + filelock anaconda/pkgs/main/win-64::filelock-3.13.1-py310haa95532_0 + fsspec anaconda/pkgs/main/win-64::fsspec-2024.6.1-py310haa95532_0 + gmpy2 anaconda/pkgs/main/win-64::gmpy2-2.1.2-py310h7f96b67_0 + jinja2 anaconda/pkgs/main/win-64::jinja2-3.1.4-py310haa95532_1 + libuv anaconda/pkgs/main/win-64::libuv-1.48.0-h827c3e9_0 + mkl-service anaconda/pkgs/main/win-64::mkl-service-2.4.0-py310h2bbff1b_1 + mkl_fft anaconda/pkgs/main/win-64::mkl_fft-1.3.11-py310h827c3e9_0 + mkl_random anaconda/pkgs/main/win-64::mkl_random-1.2.8-py310hc64d2fc_0 + mpc anaconda/pkgs/main/win-64::mpc-1.1.0-h7edee0f_1 + mpfr anaconda/pkgs/main/win-64::mpfr-4.0.2-h62dcd97_1 + mpir anaconda/pkgs/main/win-64::mpir-3.0.0-hec2e145_1 + mpmath anaconda/pkgs/main/win-64::mpmath-1.3.0-py310haa95532_0 + networkx anaconda/pkgs/main/win-64::networkx-3.3-py310haa95532_0 + numpy-base anaconda/pkgs/main/win-64::numpy-base-1.26.4-py310h65a83cf_0 + pytorch anaconda/pkgs/main/win-64::pytorch-2.3.1-cpu_py310h0ce1571_0 + pytorch-cpu anaconda/pkgs/main/win-64::pytorch-cpu-2.3.1-haa95532_0 + sympy anaconda/pkgs/main/win-64::sympy-1.13.3-py310haa95532_0 + +The following packages will be REMOVED: + + libblas-3.9.0-26_win64_mkl + libcblas-3.9.0-26_win64_mkl + liblapack-3.9.0-26_win64_mkl + +The following packages will be UPDATED: + + numpy anaconda/cloud/conda-forge::numpy-1.2~ --> anaconda/pkgs/main::numpy-1.26.4-py310h055cbcc_0 + +The following packages will be SUPERSEDED by a higher-priority channel: + + intel-openmp anaconda/cloud/conda-forge::intel-ope~ --> anaconda/pkgs/main::intel-openmp-2023.1.0-h59b6b97_46320 + mkl anaconda/cloud/conda-forge::mkl-2024.~ --> anaconda/pkgs/main::mkl-2023.1.0-h6b88ed4_46358 + + + +Downloading and Extracting Packages: + +Preparing transaction: done +Verifying transaction: done +Executing transaction: done +Looking in indexes: https://mirrors.aliyun.com/pypi/simple/ +Requirement already satisfied: pip in c:\users\eleven\desktop\easyvtuber-main\envs\miniconda3\envs\ezvtb_rt_venv\lib\site-packages (24.2) +Collecting pip + Downloading https://mirrors.aliyun.com/pypi/packages/ef/7d/500c9ad20238fcfcb4cb9243eede163594d7020ce87bd9610c9e02771876/pip-24.3.1-py3-none-any.whl (1.8 MB) + ---------------------------------------- 1.8/1.8 MB 1.4 MB/s eta 0:00:00 +Requirement already satisfied: wheel in c:\users\eleven\desktop\easyvtuber-main\envs\miniconda3\envs\ezvtb_rt_venv\lib\site-packages (0.44.0) +Collecting wheel + Downloading https://mirrors.aliyun.com/pypi/packages/0b/2c/87f3254fd8ffd29e4c02732eee68a83a1d3c346ae39bc6822dcbcb697f2b/wheel-0.45.1-py3-none-any.whl (72 kB) +Installing collected packages: wheel, pip + Attempting uninstall: wheel + Found existing installation: wheel 0.44.0 + Uninstalling wheel-0.44.0: + Successfully uninstalled wheel-0.44.0 + Attempting uninstall: pip + Found existing installation: pip 24.2 + Uninstalling pip-24.2: + Successfully uninstalled pip-24.2 +Successfully installed pip-24.3.1 wheel-0.45.1 +Looking in indexes: https://mirrors.aliyun.com/pypi/simple/ +Collecting nvidia-cudnn-cu12 + Downloading https://mirrors.aliyun.com/pypi/packages/07/1d/3c7055e1d189f55ed25183f50e1d82f477024289f925df22ac89c0f4d387/nvidia_cudnn_cu12-9.6.0.74-py3-none-win_amd64.whl (502.8 MB) + ---------------------------------------- 502.8/502.8 MB 6.3 MB/s eta 0:00:00 +Collecting nvidia-cublas-cu12 (from nvidia-cudnn-cu12) + Downloading https://mirrors.aliyun.com/pypi/packages/84/f7/985e9bdbe3e0ac9298fcc8cfa51a392862a46a0ffaccbbd56939b62a9c83/nvidia_cublas_cu12-12.6.4.1-py3-none-win_amd64.whl (434.5 MB) + ---------------------------------------- 434.5/434.5 MB 6.2 MB/s eta 0:00:00 +Installing collected packages: nvidia-cublas-cu12, nvidia-cudnn-cu12 +Successfully installed nvidia-cublas-cu12-12.6.4.1 nvidia-cudnn-cu12-9.6.0.74 +Looking in indexes: https://pypi.org/simple, https://pypi.nvidia.com +Collecting tensorrt_cu12_libs==10.6.0 + Downloading https://pypi.nvidia.com/tensorrt-cu12-libs/tensorrt_cu12_libs-10.6.0-py2.py3-none-win_amd64.whl (1075.0 MB) + ---------------------------------------- 1.1/1.1 GB 8.2 MB/s eta 0:00:00 +Collecting tensorrt_cu12_bindings==10.6.0 + Downloading https://pypi.nvidia.com/tensorrt-cu12-bindings/tensorrt_cu12_bindings-10.6.0-cp310-none-win_amd64.whl (814 kB) + ---------------------------------------- 814.1/814.1 kB 12.0 MB/s eta 0:00:00 +Collecting tensorrt==10.6.0 + Downloading https://pypi.nvidia.com/tensorrt/tensorrt-10.6.0.tar.gz (16 kB) + Preparing metadata (setup.py) ... done +Collecting nvidia-cuda-runtime-cu12 (from tensorrt_cu12_libs==10.6.0) + Downloading https://pypi.nvidia.com/nvidia-cuda-runtime-cu12/nvidia_cuda_runtime_cu12-12.6.77-py3-none-win_amd64.whl (891 kB) + ---------------------------------------- 891.8/891.8 kB 10.2 MB/s eta 0:00:00 +Collecting tensorrt-cu12==10.6.0 (from tensorrt==10.6.0) + Downloading https://pypi.nvidia.com/tensorrt-cu12/tensorrt-cu12-10.6.0.tar.gz (18 kB) + Preparing metadata (setup.py) ... done +Building wheels for collected packages: tensorrt, tensorrt-cu12 + Building wheel for tensorrt (setup.py) ... done + Created wheel for tensorrt: filename=tensorrt-10.6.0-py2.py3-none-any.whl size=16345 sha256=3fc94cd935389ea9535e279dc81bdf07b97da751150f64a1f45c39ba8b5c2663 + Stored in directory: c:\users\eleven\appdata\local\pip\cache\wheels\42\bd\86\598d26504705081ad82ae23a08a869322ee5a8bdb410d5d298 + Building wheel for tensorrt-cu12 (setup.py) ... done + Created wheel for tensorrt-cu12: filename=tensorrt_cu12-10.6.0-py2.py3-none-any.whl size=17568 sha256=1856795aec69a0f6767ab62c0962d586ea16c3301c0f7040cde242c5f92f9abc + Stored in directory: c:\users\eleven\appdata\local\pip\cache\wheels\13\99\06\326119ed088736cbd6ded866ada42bb29bbb67bed34903e57b +Successfully built tensorrt tensorrt-cu12 +Installing collected packages: tensorrt_cu12_bindings, tensorrt-cu12, nvidia-cuda-runtime-cu12, tensorrt_cu12_libs, tensorrt +Successfully installed nvidia-cuda-runtime-cu12-12.6.77 tensorrt-10.6.0 tensorrt-cu12-10.6.0 tensorrt_cu12_bindings-10.6.0 tensorrt_cu12_libs-10.6.0 +Looking in indexes: https://mirrors.aliyun.com/pypi/simple/ +Requirement already satisfied: numpy==1.26.4 in c:\users\eleven\desktop\easyvtuber-main\envs\miniconda3\envs\ezvtb_rt_venv\lib\site-packages (from -r requirements.txt (line 1)) (1.26.4) +Collecting scipy (from -r requirements.txt (line 2)) + Downloading https://mirrors.aliyun.com/pypi/packages/51/8c/c2d371111961f737ae08881f654cf54eca796c42ec0429add2a06df97049/scipy-1.15.0-cp310-cp310-win_amd64.whl (43.9 MB) + ---------------------------------------- 43.9/43.9 MB 6.5 MB/s eta 0:00:00 +Collecting matplotlib (from -r requirements.txt (line 3)) + Downloading https://mirrors.aliyun.com/pypi/packages/5a/85/b9a54d64585a6b8737a78a61897450403c30f39e0bd3214270bb0b96f002/matplotlib-3.10.0-cp310-cp310-win_amd64.whl (8.0 MB) + ---------------------------------------- 8.0/8.0 MB 6.7 MB/s eta 0:00:00 +Collecting protobuf==3.20.1 (from -r requirements.txt (line 4)) + Downloading https://mirrors.aliyun.com/pypi/packages/c0/9c/bb88091287418ae1cf8af2bb9ed9710748a562b9abc227e4884d687a8650/protobuf-3.20.1-cp310-cp310-win_amd64.whl (903 kB) + ---------------------------------------- 903.8/903.8 kB 6.8 MB/s eta 0:00:00 +Collecting pynput==1.7.6 (from -r requirements.txt (line 5)) + Downloading https://mirrors.aliyun.com/pypi/packages/02/27/4de87850ff87c8dcecaaf8d27f28cec89ef17eeb6938f250449cb2635e06/pynput-1.7.6-py2.py3-none-any.whl (89 kB) +Collecting mediapipe (from -r requirements.txt (line 6)) + Downloading https://mirrors.aliyun.com/pypi/packages/f6/7a/448885fc960a972d8bf1939602919fdaa8de17f505b6c50782a074e7b70e/mediapipe-0.10.20-cp310-cp310-win_amd64.whl (51.0 MB) + ---------------------------------------- 51.0/51.0 MB 6.2 MB/s eta 0:00:00 +Collecting opencv_python==4.5.5.64 (from -r requirements.txt (line 7)) + Downloading https://mirrors.aliyun.com/pypi/packages/48/c3/798bd7b8f78430f82ec0660b753106717e4e4bb8032ce56f77d8577cb24b/opencv_python-4.5.5.64-cp36-abi3-win_amd64.whl (35.4 MB) + ---------------------------------------- 35.4/35.4 MB 6.3 MB/s eta 0:00:00 +Collecting Pillow==9.1.0 (from -r requirements.txt (line 8)) + Downloading https://mirrors.aliyun.com/pypi/packages/a8/df/1177786a2d1c0bf732ba6d5f05a2fa40f016e81e1c16d62f1101e35d271e/Pillow-9.1.0-cp310-cp310-win_amd64.whl (3.3 MB) + ---------------------------------------- 3.3/3.3 MB 6.4 MB/s eta 0:00:00 +Collecting pyanime4k==2.5.2 (from -r requirements.txt (line 9)) + Downloading https://mirrors.aliyun.com/pypi/packages/81/af/25a4a2e9a3d46820c016f6bb2cc235d953f104546a079c8c67bf09548734/pyanime4k-2.5.2-py3-none-win_amd64.whl (35.4 MB) + ---------------------------------------- 35.4/35.4 MB 6.3 MB/s eta 0:00:00 +Collecting pyvirtualcam==0.9.1 (from -r requirements.txt (line 10)) + Downloading https://mirrors.aliyun.com/pypi/packages/ca/59/e245e2d9832c6cefd212133d5c1ef988e66e7ccd93f4381017e844949334/pyvirtualcam-0.9.1-cp310-cp310-win_amd64.whl (158 kB) +Collecting pyopengl (from -r requirements.txt (line 11)) + Downloading https://mirrors.aliyun.com/pypi/packages/99/48/00e31747821d3fc56faddd00a4725454d1e694a8b67d715cf20f531506a5/PyOpenGL-3.1.7-py3-none-any.whl (2.4 MB) + ---------------------------------------- 2.4/2.4 MB 6.3 MB/s eta 0:00:00 +Collecting onnx (from -r requirements.txt (line 12)) + Downloading https://mirrors.aliyun.com/pypi/packages/0e/d3/d26ebf590a65686dde6b27fef32493026c5be9e42083340d947395f93405/onnx-1.17.0-cp310-cp310-win_amd64.whl (14.5 MB) + ---------------------------------------- 14.5/14.5 MB 6.0 MB/s eta 0:00:00 +Collecting onnxruntime-directml (from -r requirements.txt (line 13)) + Downloading https://mirrors.aliyun.com/pypi/packages/3c/4f/f433239b05304aa9af0217da20508abbbcec1dcd58ee821e3dab8939ecfe/onnxruntime_directml-1.20.1-cp310-cp310-win_amd64.whl (22.8 MB) + ---------------------------------------- 22.8/22.8 MB 6.3 MB/s eta 0:00:00 +Collecting turbojpeg (from -r requirements.txt (line 14)) + Downloading https://mirrors.aliyun.com/pypi/packages/a0/06/1a6ca4e7eb9c8342e5991093a36dcde37033a8f8da58dcf680282bb2b733/turbojpeg-0.0.2-cp310-cp310-win_amd64.whl (439 kB) +Collecting tqdm (from -r requirements.txt (line 15)) + Downloading https://mirrors.aliyun.com/pypi/packages/d0/30/dc54f88dd4a2b5dc8a0279bdd7270e735851848b762aeb1c1184ed1f6b14/tqdm-4.67.1-py3-none-any.whl (78 kB) +Collecting six (from pynput==1.7.6->-r requirements.txt (line 5)) + Downloading https://mirrors.aliyun.com/pypi/packages/b7/ce/149a00dd41f10bc29e5921b496af8b574d8413afcd5e30dfa0ed46c2cc5e/six-1.17.0-py2.py3-none-any.whl (11 kB) +Collecting ffmpeg-python>=0.2.0 (from pyanime4k==2.5.2->-r requirements.txt (line 9)) + Downloading https://mirrors.aliyun.com/pypi/packages/d7/0c/56be52741f75bad4dc6555991fabd2e07b432d333da82c11ad701123888a/ffmpeg_python-0.2.0-py3-none-any.whl (25 kB) +Collecting contourpy>=1.0.1 (from matplotlib->-r requirements.txt (line 3)) + Downloading https://mirrors.aliyun.com/pypi/packages/e0/82/c372be3fc000a3b2005061ca623a0d1ecd2eaafb10d9e883a2fc8566e951/contourpy-1.3.1-cp310-cp310-win_amd64.whl (218 kB) +Collecting cycler>=0.10 (from matplotlib->-r requirements.txt (line 3)) + Downloading https://mirrors.aliyun.com/pypi/packages/e7/05/c19819d5e3d95294a6f5947fb9b9629efb316b96de511b418c53d245aae6/cycler-0.12.1-py3-none-any.whl (8.3 kB) +Collecting fonttools>=4.22.0 (from matplotlib->-r requirements.txt (line 3)) + Downloading https://mirrors.aliyun.com/pypi/packages/19/d1/4dcd865360fb2c499749a913fe80e41c26e8ae18629d87dfffa3de27e831/fonttools-4.55.3-cp310-cp310-win_amd64.whl (2.2 MB) + ---------------------------------------- 2.2/2.2 MB 6.3 MB/s eta 0:00:00 +Collecting kiwisolver>=1.3.1 (from matplotlib->-r requirements.txt (line 3)) + Downloading https://mirrors.aliyun.com/pypi/packages/98/99/0dd05071654aa44fe5d5e350729961e7bb535372935a45ac89a8924316e6/kiwisolver-1.4.8-cp310-cp310-win_amd64.whl (71 kB) +Collecting packaging>=20.0 (from matplotlib->-r requirements.txt (line 3)) + Downloading https://mirrors.aliyun.com/pypi/packages/88/ef/eb23f262cca3c0c4eb7ab1933c3b1f03d021f2c48f54763065b6f0e321be/packaging-24.2-py3-none-any.whl (65 kB) +Collecting pyparsing>=2.3.1 (from matplotlib->-r requirements.txt (line 3)) + Downloading https://mirrors.aliyun.com/pypi/packages/1c/a7/c8a2d361bf89c0d9577c934ebb7421b25dc84bf3a8e3ac0a40aed9acc547/pyparsing-3.2.1-py3-none-any.whl (107 kB) +Collecting python-dateutil>=2.7 (from matplotlib->-r requirements.txt (line 3)) + Downloading https://mirrors.aliyun.com/pypi/packages/ec/57/56b9bcc3c9c6a792fcbaf139543cee77261f3651ca9da0c93f5c1221264b/python_dateutil-2.9.0.post0-py2.py3-none-any.whl (229 kB) +Collecting absl-py (from mediapipe->-r requirements.txt (line 6)) + Downloading https://mirrors.aliyun.com/pypi/packages/a2/ad/e0d3c824784ff121c03cc031f944bc7e139a8f1870ffd2845cc2dd76f6c4/absl_py-2.1.0-py3-none-any.whl (133 kB) +Collecting attrs>=19.1.0 (from mediapipe->-r requirements.txt (line 6)) + Downloading https://mirrors.aliyun.com/pypi/packages/89/aa/ab0f7891a01eeb2d2e338ae8fecbe57fcebea1a24dbb64d45801bfab481d/attrs-24.3.0-py3-none-any.whl (63 kB) +Collecting flatbuffers>=2.0 (from mediapipe->-r requirements.txt (line 6)) + Downloading https://mirrors.aliyun.com/pypi/packages/fb/b4/31c461eef98b96b8ab736d97274548eaf2b2e349bf09e4de3902f7d53084/flatbuffers-24.12.23-py2.py3-none-any.whl (30 kB) +Collecting jax (from mediapipe->-r requirements.txt (line 6)) + Downloading https://mirrors.aliyun.com/pypi/packages/22/49/b4418a7a892c0dd64442bbbeef54e1cdfe722dfc5a7bf0d611d3f5f90e99/jax-0.4.38-py3-none-any.whl (2.2 MB) + ---------------------------------------- 2.2/2.2 MB 6.3 MB/s eta 0:00:00 +Collecting jaxlib (from mediapipe->-r requirements.txt (line 6)) + Downloading https://mirrors.aliyun.com/pypi/packages/0d/04/120c4caac6151f7297fedf9dd776362aa2d417d3f87bda826050b4da45e8/jaxlib-0.4.38-cp310-cp310-win_amd64.whl (64.2 MB) + ---------------------------------------- 64.2/64.2 MB 6.2 MB/s eta 0:00:00 +WARNING: Retrying (Retry(total=4, connect=None, read=None, redirect=None, status=None)) after connection broken by 'ReadTimeoutError("HTTPSConnectionPool(host='mirrors.aliyun.com', port=443): Read timed out. (read timeout=15)")': /pypi/simple/opencv-contrib-python/ +Collecting opencv-contrib-python (from mediapipe->-r requirements.txt (line 6)) + Downloading https://mirrors.aliyun.com/pypi/packages/a7/9e/7110d2c5d543ab03b9581dbb1f8e2429863e44e0c9b4960b766f230c1279/opencv_contrib_python-4.10.0.84-cp37-abi3-win_amd64.whl (45.5 MB) + ---------------------------------------- 45.5/45.5 MB 6.3 MB/s eta 0:00:00 +INFO: pip is looking at multiple versions of mediapipe to determine which version is compatible with other requirements. This could take a while. +Collecting mediapipe (from -r requirements.txt (line 6)) + Downloading https://mirrors.aliyun.com/pypi/packages/d0/ca/4a32e95f1767467f7c461d7d5b4b506faebb1e30e35c1031c1b31cf82846/mediapipe-0.10.18-cp310-cp310-win_amd64.whl (50.9 MB) + ---------------------------------------- 50.9/50.9 MB 6.3 MB/s eta 0:00:00 + Downloading https://mirrors.aliyun.com/pypi/packages/ea/1b/79e1ddc041012f5e5a7d60cbfd8cfe6037c5c87f4e77b4fdc0f9dfa2d44e/mediapipe-0.10.14-cp310-cp310-win_amd64.whl (50.8 MB) + ---------------------------------------- 50.8/50.8 MB 6.3 MB/s eta 0:00:00 + Downloading https://mirrors.aliyun.com/pypi/packages/bb/b7/ffcdd19de3bf443864d0b0fed2b102275b96cbfbc1ddcf96ad811bbc1af0/mediapipe-0.10.13-cp310-cp310-win_amd64.whl (50.8 MB) + ---------------------------------------- 50.8/50.8 MB 6.4 MB/s eta 0:00:00 + Downloading https://mirrors.aliyun.com/pypi/packages/12/50/9c24e158350d3f93be669db291fb452f21a25d874c94c5758374be82fff1/mediapipe-0.10.11-cp310-cp310-win_amd64.whl (50.8 MB) + ---------------------------------------- 50.8/50.8 MB 6.4 MB/s eta 0:00:00 +Collecting sounddevice>=0.4.4 (from mediapipe->-r requirements.txt (line 6)) + Downloading https://mirrors.aliyun.com/pypi/packages/af/9b/15217b04f3b36d30de55fef542389d722de63f1ad81f9c72d8afc98cb6ab/sounddevice-0.5.1-py3-none-win_amd64.whl (363 kB) +INFO: pip is looking at multiple versions of onnx to determine which version is compatible with other requirements. This could take a while. +Collecting onnx (from -r requirements.txt (line 12)) + Downloading https://mirrors.aliyun.com/pypi/packages/0c/e4/1bc3ae56e6581587926a50a5c9dce3cfacf510e592e2512c7f5f2a9a4859/onnx-1.16.2-cp310-cp310-win_amd64.whl (14.4 MB) + ---------------------------------------- 14.4/14.4 MB 6.4 MB/s eta 0:00:00 + Downloading https://mirrors.aliyun.com/pypi/packages/9c/7c/40fbebcb30f0fb6a773ca36e3a9dd4bdccd4b15455ef6c21a335781db78f/onnx-1.16.1-cp310-cp310-win_amd64.whl (14.4 MB) + ---------------------------------------- 14.4/14.4 MB 6.3 MB/s eta 0:00:00 + Downloading https://mirrors.aliyun.com/pypi/packages/3b/e2/471ff83b3862967791d67f630000afce038756afbdf0665a3d767677c851/onnx-1.16.0-cp310-cp310-win_amd64.whl (14.4 MB) + ---------------------------------------- 14.4/14.4 MB 6.3 MB/s eta 0:00:00 + Downloading https://mirrors.aliyun.com/pypi/packages/1b/43/6e84bf35a9201fb344b1a98edf7095c4fa1cf1478dfb6638d7b77f5475e6/onnx-1.15.0-cp310-cp310-win_amd64.whl (14.3 MB) + ---------------------------------------- 14.3/14.3 MB 6.3 MB/s eta 0:00:00 + Downloading https://mirrors.aliyun.com/pypi/packages/00/36/e7a7e7a85564e7d409e4e8addfa11d41015d2190bfff30064771e7c21ca0/onnx-1.14.1-cp310-cp310-win_amd64.whl (13.3 MB) + ---------------------------------------- 13.3/13.3 MB 6.2 MB/s eta 0:00:00 + Downloading https://mirrors.aliyun.com/pypi/packages/22/5c/46298252ea9f92b6b94184e8f001e575f2c346a22011498110fd032fc921/onnx-1.14.0-cp310-cp310-win_amd64.whl (13.3 MB) + ---------------------------------------- 13.3/13.3 MB 6.2 MB/s eta 0:00:00 + Downloading https://mirrors.aliyun.com/pypi/packages/bb/cd/e2e63b1f0922dd45f6941a3422ccdb400360955064654a1a3a20e6a613fb/onnx-1.13.1-cp310-cp310-win_amd64.whl (12.2 MB) + ---------------------------------------- 12.2/12.2 MB 6.2 MB/s eta 0:00:00 +INFO: pip is still looking at multiple versions of onnx to determine which version is compatible with other requirements. This could take a while. + Downloading https://mirrors.aliyun.com/pypi/packages/97/e1/bdb42dde046507250083862520428901af3f5182faac50d3df5f13a4ff3d/onnx-1.13.0-cp310-cp310-win_amd64.whl (12.2 MB) + ---------------------------------------- 12.2/12.2 MB 6.5 MB/s eta 0:00:00 + Downloading https://mirrors.aliyun.com/pypi/packages/b0/f0/5e383f64293857e86c300cb70010cce1ca6f038f1563b85501d30404ad2a/onnx-1.12.0-cp310-cp310-win_amd64.whl (11.5 MB) + ---------------------------------------- 11.5/11.5 MB 6.2 MB/s eta 0:00:00 +Requirement already satisfied: typing-extensions>=3.6.2.1 in c:\users\eleven\desktop\easyvtuber-main\envs\miniconda3\envs\ezvtb_rt_venv\lib\site-packages (from onnx->-r requirements.txt (line 12)) (4.12.2) +Collecting coloredlogs (from onnxruntime-directml->-r requirements.txt (line 13)) + Downloading https://mirrors.aliyun.com/pypi/packages/a7/06/3d6badcf13db419e25b07041d9c7b4a2c331d3f4e7134445ec5df57714cd/coloredlogs-15.0.1-py2.py3-none-any.whl (46 kB) +Requirement already satisfied: sympy in c:\users\eleven\desktop\easyvtuber-main\envs\miniconda3\envs\ezvtb_rt_venv\lib\site-packages (from onnxruntime-directml->-r requirements.txt (line 13)) (1.13.3) +Collecting colorama (from tqdm->-r requirements.txt (line 15)) + Downloading https://mirrors.aliyun.com/pypi/packages/d1/d6/3965ed04c63042e047cb6a3e6ed1a63a35087b6a609aa3a15ed8ac56c221/colorama-0.4.6-py2.py3-none-any.whl (25 kB) +Collecting future (from ffmpeg-python>=0.2.0->pyanime4k==2.5.2->-r requirements.txt (line 9)) + Downloading https://mirrors.aliyun.com/pypi/packages/da/71/ae30dadffc90b9006d77af76b393cb9dfbfc9629f339fc1574a1c52e6806/future-1.0.0-py3-none-any.whl (491 kB) +Collecting CFFI>=1.0 (from sounddevice>=0.4.4->mediapipe->-r requirements.txt (line 6)) + Downloading https://mirrors.aliyun.com/pypi/packages/d1/b6/0b0f5ab93b0df4acc49cae758c81fe4e5ef26c3ae2e10cc69249dfd8b3ab/cffi-1.17.1-cp310-cp310-win_amd64.whl (181 kB) +Collecting humanfriendly>=9.1 (from coloredlogs->onnxruntime-directml->-r requirements.txt (line 13)) + Downloading https://mirrors.aliyun.com/pypi/packages/f0/0f/310fb31e39e2d734ccaa2c0fb981ee41f7bd5056ce9bc29b2248bd569169/humanfriendly-10.0-py2.py3-none-any.whl (86 kB) +Collecting ml_dtypes>=0.4.0 (from jax->mediapipe->-r requirements.txt (line 6)) + Downloading https://mirrors.aliyun.com/pypi/packages/6b/7e/bc54ae885e4d702e60a4bf50aa9066ff35e9c66b5213d11091f6bffb3036/ml_dtypes-0.5.1-cp310-cp310-win_amd64.whl (209 kB) +Collecting opt_einsum (from jax->mediapipe->-r requirements.txt (line 6)) + Downloading https://mirrors.aliyun.com/pypi/packages/23/cd/066e86230ae37ed0be70aae89aabf03ca8d9f39c8aea0dec8029455b5540/opt_einsum-3.4.0-py3-none-any.whl (71 kB) +Requirement already satisfied: mpmath<1.4,>=1.1.0 in c:\users\eleven\desktop\easyvtuber-main\envs\miniconda3\envs\ezvtb_rt_venv\lib\site-packages (from sympy->onnxruntime-directml->-r requirements.txt (line 13)) (1.3.0) +Collecting pycparser (from CFFI>=1.0->sounddevice>=0.4.4->mediapipe->-r requirements.txt (line 6)) + Downloading https://mirrors.aliyun.com/pypi/packages/13/a3/a812df4e2dd5696d1f351d58b8fe16a405b234ad2886a0dab9183fb78109/pycparser-2.22-py3-none-any.whl (117 kB) +Collecting pyreadline3 (from humanfriendly>=9.1->coloredlogs->onnxruntime-directml->-r requirements.txt (line 13)) + Downloading https://mirrors.aliyun.com/pypi/packages/5a/dc/491b7661614ab97483abf2056be1deee4dc2490ecbf7bff9ab5cdbac86e1/pyreadline3-3.5.4-py3-none-any.whl (83 kB) +Installing collected packages: pyopengl, flatbuffers, turbojpeg, six, scipy, pyvirtualcam, pyreadline3, pyparsing, pycparser, protobuf, Pillow, packaging, opt_einsum, opencv_python, opencv-contrib-python, ml_dtypes, kiwisolver, future, fonttools, cycler, contourpy, colorama, attrs, absl-py, tqdm, python-dateutil, pynput, onnx, jaxlib, humanfriendly, ffmpeg-python, CFFI, sounddevice, pyanime4k, matplotlib, jax, coloredlogs, onnxruntime-directml, mediapipe +Successfully installed CFFI-1.17.1 Pillow-9.1.0 absl-py-2.1.0 attrs-24.3.0 colorama-0.4.6 coloredlogs-15.0.1 contourpy-1.3.1 cycler-0.12.1 ffmpeg-python-0.2.0 flatbuffers-24.12.23 fonttools-4.55.3 future-1.0.0 humanfriendly-10.0 jax-0.4.38 jaxlib-0.4.38 kiwisolver-1.4.8 matplotlib-3.10.0 mediapipe-0.10.11 ml_dtypes-0.5.1 onnx-1.12.0 onnxruntime-directml-1.20.1 opencv-contrib-python-4.10.0.84 opencv_python-4.5.5.64 opt_einsum-3.4.0 packaging-24.2 protobuf-3.20.1 pyanime4k-2.5.2 pycparser-2.22 pynput-1.7.6 pyopengl-3.1.7 pyparsing-3.2.1 pyreadline3-3.5.4 python-dateutil-2.9.0.post0 pyvirtualcam-0.9.1 scipy-1.15.0 six-1.17.0 sounddevice-0.5.1 tqdm-4.67.1 turbojpeg-0.0.2 +Press any key to continue . . . \ No newline at end of file diff --git a/ezvtb_rt_interface.py b/ezvtb_rt_interface.py index e8bef72..a73ece7 100644 --- a/ezvtb_rt_interface.py +++ b/ezvtb_rt_interface.py @@ -34,6 +34,7 @@ def get_core( model_cache:bool = True,#Only works for tensorrt model_vram_cache:bool = True, #Only works for tensorrt model_cache_size:float = 1.0, #unit of GigaBytes, only works for tensorrt + model_use_eyebrow:bool = True, #RIFE interpolation setting use_interpolation:bool = True, interpolation_scale:int = 2, @@ -106,10 +107,10 @@ def get_core( else: if model_vram_cache: from ezvtb_rt.tha import THACoreCachedVRAM - tha = THACoreCachedVRAM(tha_model_dir, model_cache_size) + tha = THACoreCachedVRAM(tha_model_dir, model_cache_size, model_use_eyebrow) else: from ezvtb_rt.tha import THACoreCachedRAM - tha = THACoreCachedRAM(tha_model_dir, model_cache_size) + tha = THACoreCachedRAM(tha_model_dir, model_cache_size, model_use_eyebrow) rife = None if use_interpolation: from ezvtb_rt.rife import RIFECoreLinked @@ -122,7 +123,7 @@ def get_core( core = Core(tha, cacher, sr, rife) else: #Use directml from ezvtb_rt.core_ort import CoreORT - 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) + 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, use_eyebrow = model_use_eyebrow) return core diff --git a/ezvtuber-rt b/ezvtuber-rt index c841f5f..09de205 160000 --- a/ezvtuber-rt +++ b/ezvtuber-rt @@ -1 +1 @@ -Subproject commit c841f5f4378e81029b4fce0b7a0e0ac17fea9f43 +Subproject commit 09de205f3b5b53e984689d78a22b805a2592d18c diff --git a/launcher2.py b/launcher2.py index 4b95380..3a04997 100644 --- a/launcher2.py +++ b/launcher2.py @@ -9,7 +9,7 @@ import sys ctypes.windll.shcore.SetProcessDpiAwareness(1) p = None cache_simplify_map = { - 'Off':0, + 'OFF':0, 'Low':1, 'Medium':2, 'High':3, @@ -20,7 +20,7 @@ cache_simplify_map = { cache_simplify_quality_map = { - 'Off':100, + 'OFF':100, 'Low':99, 'Medium':95, 'High':90, diff --git a/main.py b/main.py index 9fcb36f..5871aae 100644 --- a/main.py +++ b/main.py @@ -352,6 +352,7 @@ class ModelClientProcess(Process): model_cache=args.model_cache, model_vram_cache=args.model_vram_cache, model_cache_size=args.max_gpu_cache_len, + model_use_eyebrow=args.eyebrow, use_interpolation=args.use_interpolation, interpolation_scale=args.interpolation_scale, @@ -886,7 +887,7 @@ def main(): need_a_copy = True - postprocessed_image = model_output.copy() + postprocessed_image = model_output if args.perf == 'main': print("Compute", time.perf_counter() - tic) tic = time.perf_counter() diff --git a/requirements.txt b/requirements.txt index be68fc0..9bd7383 100644 --- a/requirements.txt +++ b/requirements.txt @@ -13,4 +13,5 @@ wxpython onnx onnxruntime-directml turbojpeg -tqdm \ No newline at end of file +tqdm +torch \ No newline at end of file