terra-cognita-frugal-ai / requirements.txt
kiurtis's picture
maybe fix the gpu onnx thing
70f4ec6
raw
history blame
456 Bytes
fastapi>=0.68.0
uvicorn>=0.15.0
codecarbon>=2.3.1
datasets>=2.14.0
scikit-learn>=1.0.2
pydantic>=1.10.0
python-dotenv>=1.0.0
gradio>=4.0.0
requests>=2.31.0
librosa==0.10.2.post1
numpy==1.26.4
ultralytics==8.3.68
ultralytics-thop==2.0.14
#opencv-python==4.11.0.86
python-dotenv==1.0.0
matplotlib==3.8.1
onnx==1.17.0
tensorrt==10.8.0.43
tensorrt_cu12==10.8.0.43
tensorrt_cu12_bindings==10.8.0.43
tensorrt_cu12_libs==10.8.0.43
optimum[onnxruntime-gpu]==1.24.0