-
-
Notifications
You must be signed in to change notification settings - Fork 6.9k
2. Acceleration
Henry Ruhs edited this page Aug 18, 2023
·
18 revisions
-
Install CUDA Toolkit 11.8 and cuDNN for Cuda 11.x
-
Install dependencies:
pip uninstall onnxruntime onnxruntime-gpu
pip install onnxruntime-gpu==1.15.1
- Usage in case the provider is available:
python run.py --execution-provider cuda
- Install dependencies:
pip uninstall onnxruntime onnxruntime-silicon
pip install onnxruntime-silicon==1.13.1
- Usage in case the provider is available:
python run.py --execution-provider coreml
1.Install dependencies:
pip uninstall onnxruntime onnxruntime-coreml
pip install onnxruntime-coreml==1.13.1
- Usage in case the provider is available:
python run.py --execution-provider coreml
- Install dependencies:
pip uninstall onnxruntime onnxruntime-directml
pip install onnxruntime-directml==1.15.1
- Usage in case the provider is available:
python run.py --execution-provider dml
- Install dependencies:
pip uninstall onnxruntime onnxruntime-openvino
pip install onnxruntime-openvino==1.15.0
- Usage in case the provider is available:
python run.py --execution-provider openvino