Compare commits

..

1 Commits

Author SHA1 Message Date
Uma Maheshwar Reddy Thipparthi 0328245e0e
Merge 0cc4a2216f into ec6d7d2995 2025-06-02 23:00:57 +05:30
1 changed files with 3 additions and 4 deletions

View File

@ -187,14 +187,13 @@ pip install -r requirements.txt
**CUDA Execution Provider (Nvidia)** **CUDA Execution Provider (Nvidia)**
1. Install [CUDA Toolkit 12.8.0](https://developer.nvidia.com/cuda-12-8-0-download-archive) 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 12.x](https://developer.nvidia.com/rdp/cudnn-archive) (required for onnxruntime-gpu): 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 12.x - Download cuDNN v8.9.7 for CUDA 11.x
- Make sure the cuDNN bin directory is in your system PATH - Make sure the cuDNN bin directory is in your system PATH
3. Install dependencies: 3. Install dependencies:
```bash ```bash
pip install -U torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
pip uninstall onnxruntime onnxruntime-gpu pip uninstall onnxruntime onnxruntime-gpu
pip install onnxruntime-gpu==1.16.3 pip install onnxruntime-gpu==1.16.3
``` ```