From b4ab2cb4b1e56d3a76825a24fa1a813edc5548ff Mon Sep 17 00:00:00 2001 From: hhsadf Date: Sat, 28 Jun 2025 05:32:04 +0700 Subject: [PATCH 1/2] Update requirements_cu124.txt --- requirements_cu124.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements_cu124.txt b/requirements_cu124.txt index 5927a000..0e79a5bd 100644 --- a/requirements_cu124.txt +++ b/requirements_cu124.txt @@ -7,7 +7,7 @@ pillow==9.5.0 onnx==1.16.1 protobuf==4.23.2 psutil==6.0.0 -onnxruntime-gpu==1.20.0 +onnxruntime-gpu==1.22.0 packaging==24.1 PySide6==6.8.2.1 kornia From 3ac65d4f1e57214a1f170ca68c4644fb09f53cdc Mon Sep 17 00:00:00 2001 From: hhsadf Date: Sun, 29 Jun 2025 21:30:26 +0700 Subject: [PATCH 2/2] Add CUDA 12.8 support --- README.md | 10 ++++++++++ requirements_cu128.txt | 29 +++++++++++++++++++++++++++++ scripts/update_cu128.bat | 6 ++++++ 3 files changed, 45 insertions(+) create mode 100644 requirements_cu128.txt create mode 100644 scripts/update_cu128.bat diff --git a/README.md b/README.md index 62363e6e..87b0bf33 100644 --- a/README.md +++ b/README.md @@ -72,6 +72,11 @@ conda install -c nvidia/label/cuda-12.4.1 cuda-runtime ```sh conda install -c conda-forge cudnn ``` +#### CUDA 12.8 +```sh +conda install -c nvidia/label/cuda-12.8.0 cuda-runtime +``` +Compatible with GPUs having compute capability 9.0 or higher (e.g., RTX 5090). ### **4. Install Additional Dependencies** ```sh @@ -80,12 +85,17 @@ conda install scikit-image ```sh pip install -r requirements_cu124.txt ``` +```sh +pip install -r requirements_cu128.txt +``` ### **5. Download Models and Other Dependencies** 1. Download all the required models ```sh python download_models.py ``` +For CUDA 12.8 you can run `scripts\update_cu128.bat` to install the correct +packages and download the models in one step. 2. Download all the files from this [page](https://github.com/visomaster/visomaster-assets/releases/tag/v0.1.0_dp) and copy it to the ***dependencies/*** folder. **Note**: You do not need to download the Source code (zip) and Source code (tar.gz) files diff --git a/requirements_cu128.txt b/requirements_cu128.txt new file mode 100644 index 00000000..11f3047c --- /dev/null +++ b/requirements_cu128.txt @@ -0,0 +1,29 @@ +--extra-index-url https://download.pytorch.org/whl/cu128 + +numpy==1.26.4 +opencv-python==4.10.0.84 +scikit-image==0.21.0 +pillow==9.5.0 +onnx==1.16.1 +protobuf==4.23.2 +psutil==6.0.0 +onnxruntime-gpu==1.22.0 +packaging==24.1 +PySide6==6.8.2.1 +kornia +torch==2.4.1+cu128 +torchvision==0.19.1+cu128 +torchaudio==2.4.1+cu128 +tensorrt==10.6.0 --extra-index-url https://pypi.nvidia.com +tensorrt-cu12_libs==10.6.0 +tensorrt-cu12_bindings==10.6.0 +tqdm +ftfy +regex +pyvirtualcam==0.11.1 +numexpr +onnxsim +requests +pyqt-toast-notification==1.3.2 +qdarkstyle +pyqtdarktheme diff --git a/scripts/update_cu128.bat b/scripts/update_cu128.bat new file mode 100644 index 00000000..0dbc56f4 --- /dev/null +++ b/scripts/update_cu128.bat @@ -0,0 +1,6 @@ +@echo off +call scripts\setenv.bat +"%GIT_EXECUTABLE%" fetch origin main +"%GIT_EXECUTABLE%" reset --hard origin/main +"%PYTHON_EXECUTABLE%" -m pip install -r requirements_cu128.txt --default-timeout 100 +"%PYTHON_EXECUTABLE%" download_models.py