worlabel/ai/requirements.txt

13 lines
319 B
Plaintext
Raw Permalink Normal View History

fastapi==0.104.1
uvicorn==0.30.6
ultralytics==8.2.82
ultralytics-thop==2.0.5
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.4.0+cu121
--extra-index-url https://download.pytorch.org/whl/cu121
torchaudio==2.4.0+cu121
--extra-index-url https://download.pytorch.org/whl/cu121
torchvision==0.19.0+cu121