diff --git a/requirements.txt b/requirements.txt index 024a45f..4331515 100644 --- a/requirements.txt +++ b/requirements.txt @@ -33,6 +33,7 @@ filelock==3.16.1 # via # huggingface-hub # torch + # triton flatbuffers==24.3.25 # via onnxruntime fonttools==4.55.0 @@ -56,7 +57,7 @@ idna==3.10 # yarl importlib-resources==6.4.5 # via matplotlib -jinja2==3.1.4 +jinja2==3.1.5 # via torch kiwisolver==1.4.7 # via matplotlib @@ -96,6 +97,38 @@ numpy==1.26.4 # torchvision # ultralytics # ultralytics-thop +nvidia-cublas-cu12==12.4.5.8 + # via + # nvidia-cudnn-cu12 + # nvidia-cusolver-cu12 + # torch +nvidia-cuda-cupti-cu12==12.4.127 + # via torch +nvidia-cuda-nvrtc-cu12==12.4.127 + # via torch +nvidia-cuda-runtime-cu12==12.4.127 + # via torch +nvidia-cudnn-cu12==9.1.0.70 + # via torch +nvidia-cufft-cu12==11.2.1.3 + # via torch +nvidia-curand-cu12==10.3.5.147 + # via torch +nvidia-cusolver-cu12==11.6.1.9 + # via torch +nvidia-cusparse-cu12==12.3.1.170 + # via + # nvidia-cusolver-cu12 + # torch +nvidia-nccl-cu12==2.21.5 + # via torch +nvidia-nvjitlink-cu12==12.4.127 + # via + # nvidia-cusolver-cu12 + # nvidia-cusparse-cu12 + # torch +nvidia-nvtx-cu12==12.4.127 + # via torch onnx==1.17.0 # via -r requirements.in onnxruntime==1.19.2 @@ -207,6 +240,8 @@ tqdm==4.67.0 # pytorch-lightning # rapidocr-onnxruntime # ultralytics +triton==3.1.0 + # via torch typing-extensions==4.12.2 # via # huggingface-hub