Skip to content

2. Acceleration

Henry Ruhs edited this page Jun 21, 2023 · 18 revisions

CUDA Execution Provider (Nvidia)

  1. Uninstall dependencies:
pip uninstall onnxruntime onnxruntime-gpu
  1. Install dependencies:
pip install onnxruntime onnxruntime-gpu
  1. Usage in case the provider is available:
python run.py --execution-provider cuda

CoreML Execution Provider (Apple)

  1. Uninstall dependencies:
pip uninstall onnxruntime onnxruntime-silicon
  1. Install dependencies:
pip install onnxruntime onnxruntime-silicon
  1. Usage in case the provider is available:
python run.py --execution-provider coreml

OpenVINO™ Execution Provider (Intel)

  1. Uninstall dependencies:
pip uninstall onnxruntime onnxruntime-openvino
  1. Install dependencies:
pip install onnxruntime onnxruntime-openvino
  1. Usage in case the provider is available:
python run.py --execution-provider openvino

ROCm Execution Provider (AMD)

  1. Build custom wheels from the onnxruntime source.

  2. Install the custom wheels.

  3. Usage in case the provider is available:

python run.py --execution-provider rocm
Clone this wiki locally