From 745d449ca62fed0d089f7abc743a7be677c7c544 Mon Sep 17 00:00:00 2001 From: Kenneth Estanislao Date: Mon, 9 Jun 2025 00:34:26 +0800 Subject: [PATCH] Update README.md support for RTX 50xx --- README.md | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 1f1624a..18efaba 100644 --- 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 ```