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 ```