Merge 2663c7eae7 into 745d449ca6

Update README.md
support for RTX 50xx
2025-06-09 00:36:10 +02:00 · 2025-06-09 00:34:26 +08:00 · 2025-05-27 22:29:24 +02:00 · 2025-05-07 14:43:53 +02:00
2 changed files with 5 additions and 3 deletions
--- a/README.md
+++ b/README.md
@ -187,13 +187,14 @@ pip install -r requirements.txt

 **CUDA Execution Provider (Nvidia)**

-1. Install [CUDA Toolkit 11.8.0](https://developer.nvidia.com/cuda-11-8-0-download-archive)
-2. Install [cuDNN v8.9.7 for CUDA 11.x](https://developer.nvidia.com/rdp/cudnn-archive) (required for onnxruntime-gpu):
-   - Download cuDNN v8.9.7 for CUDA 11.x
+1. Install [CUDA Toolkit 12.8.0](https://developer.nvidia.com/cuda-12-8-0-download-archive)
+2. Install [cuDNN v8.9.7 for CUDA 12.x](https://developer.nvidia.com/rdp/cudnn-archive) (required for onnxruntime-gpu):
+   - Download cuDNN v8.9.7 for CUDA 12.x
   - Make sure the cuDNN bin directory is in your system PATH
 3. Install dependencies:

 ```bash
+pip install -U torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
 pip uninstall onnxruntime onnxruntime-gpu
 pip install onnxruntime-gpu==1.16.3
 ```
--- a/requirements.txt
+++ b/requirements.txt
@ -19,3 +19,4 @@ onnxruntime-gpu==1.17; sys_platform != 'darwin'
 tensorflow; sys_platform != 'darwin'
 opennsfw2==0.10.2
 protobuf==4.23.2
+pygrabber==0.2
Author	SHA1	Message	Date
GuillaumeRattin	10890f4fac	Merge `2663c7eae7` into `745d449ca6`	2025-06-09 00:36:10 +02:00
Kenneth Estanislao	745d449ca6	Update README.md support for RTX 50xx	2025-06-09 00:34:26 +08:00
GuillaumeRattin	2663c7eae7	Merge branch 'hacksider:main' into main	2025-05-27 22:29:24 +02:00
GuillaumeRattin	aafeb8cf89	Add pygrabber==0.2 to requirements.txt	2025-05-07 14:43:53 +02:00