numpy
torch
matplotlib
tqdm
easydict
Pillow
opencv_python
pycocotools
onnx
tensorflow
onnx2tf
tf-keras
psutil
h5py
ml_dtypes

[:platform_system == "Linux"]
ai-edge-litert
simple_onnx_processing_tools
onnxsim-prebuilt
onnx-graphsurgeon

[cpu]
onnxruntime

[rocm]
onnxruntime-rocm

[tensorrt]
tensorrt<10.13.2.6
pycuda
onnxruntime-gpu

[tensorrt:platform_system == "Linux"]
nvidia-cublas-cu12
nvidia-cuda-cupti-cu12
nvidia-cuda-nvrtc-cu12
nvidia-cuda-runtime-cu12
nvidia-cudnn-cu12
nvidia-cufft-cu12
nvidia-curand-cu12
nvidia-cusolver-cu12
nvidia-cusparse-cu12
nvidia-nccl-cu12
nvidia-nvjitlink-cu12
nvidia-nvtx-cu12

[vino]
onnxruntime-openvino
