parent
ec6d7d2995
commit
745d449ca6
|
@ -187,13 +187,14 @@ pip install -r requirements.txt
|
||||||
|
|
||||||
**CUDA Execution Provider (Nvidia)**
|
**CUDA Execution Provider (Nvidia)**
|
||||||
|
|
||||||
1. Install [CUDA Toolkit 11.8.0](https://developer.nvidia.com/cuda-11-8-0-download-archive)
|
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 11.x](https://developer.nvidia.com/rdp/cudnn-archive) (required for onnxruntime-gpu):
|
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 11.x
|
- Download cuDNN v8.9.7 for CUDA 12.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
|
||||||
```
|
```
|
||||||
|
|
Loading…
Reference in New Issue