requirements.txt

coloredlogs==15.0.1
ConfigArgParse==1.7
Cython==3.0.11
fast-histogram==0.14
filelock==3.16.1
flatbuffers==24.12.23
fsspec==2024.12.0
humanfriendly==10.0
Jinja2==3.1.5
MarkupSafe==3.0.2
mpmath==1.3.0
networkx==3.2.1
numpy==1.26.4
nvidia-cublas-cu12==12.1.3.1
nvidia-cuda-cupti-cu12==12.1.105
nvidia-cuda-nvrtc-cu12==12.1.105
nvidia-cuda-runtime-cu12==12.1.105
nvidia-cudnn-cu12==9.1.0.70
nvidia-cufft-cu12==11.0.2.54
nvidia-curand-cu12==10.3.2.106
nvidia-cusolver-cu12==11.4.5.107
nvidia-cusparse-cu12==12.1.0.106
nvidia-nccl-cu12==2.20.5
nvidia-nvjitlink-cu12==12.6.85
nvidia-nvtx-cu12==12.1.105
onnx==1.17.0
onnxruntime==1.19.2
opencv-python==4.10.0.84
packaging==24.2
protobuf==4.25.4
psutil==6.1.1
rknn-toolkit2==2.3.0
ruamel.yaml==0.18.7
ruamel.yaml.clib==0.2.12
scipy==1.13.1
sympy==1.13.3
torch==2.4.0
tqdm==4.67.1
triton==3.0.0
typing_extensions==4.12.2