Files
aicheckv2/requirements_gpu.txt

41 lines
990 B
Plaintext

# aicheck_v2 requirements
# Usage: pip install -r requirements.txt
# API -------------------------------------------------------------------------
fastapi~=0.74.1 # fastapi基础包
pydantic~=1.9.0 # pydantic
email-validator==1.1.3
PyJWT~=2.3.0
python-multipart==0.0.5
redis==4.1.4
SQLAlchemy~=2.0.34
cryptography==44.0.2
uvicorn~=0.17.5
uvicorn[standard]
loguru~=0.6.0
xlrd~=2.0.1
bcrypt==3.2.0
pymysql==1.0.2
pynvml==12.0.0
requests-toolbelt==1.0.0
python-socketio == 5.12.1
setuptools==75.8.0
# YOLOV5 ----------------------------------------------------------------------
gitpython>=3.1.30
matplotlib==3.7.0
numpy==1.24.0 # cuda版本的numpy需要降一下版本
opencv-python>=4.1.1
pillow>=10.3.0
PyYAML>=5.3.1
requests>=2.32.2
scipy==1.13.1
thop>=0.1.1 # FLOPs computation
# torch==2.6.0+cu124 # 本地安装
# torchvision==0.21.0+cu124 # 本地安装
tqdm>=4.66.3
ultralytics==8.3.75 # https://ultralytics.com
pandas==2.2.3
seaborn==0.11.2 # 对应的这个依赖也需要降