gpu_requirements.txt

# original

albumentations==1.3.1
certifi==2023.7.22
charset-normalizer==3.2.0
colorama==0.4.6
contourpy==1.1.0
cycler==0.11.0
efficientnet-pytorch==0.7.1
filelock==3.12.2
fonttools==4.41.1
fsspec==2023.9.0
geomdl==5.3.1
gitdb==4.0.10
GitPython==3.1.32
huggingface-hub==0.17.1
idna==3.4
imageio==2.28.1
joblib==1.2.0
kiwisolver==1.4.4
lazy_loader==0.2
matplotlib==3.7.2
matplotlib-inline==0.1.6
munch==4.0.0
networkx==3.1
numpy==1.23.5
opencv-contrib-python==4.8.0.76
opencv-python==4.8.0.76
# opencv-python-headless==4.8.0.76
packaging==23.1
pandas==2.0.3
Pillow==10.0.0
pretrainedmodels==0.7.4
psutil==5.9.5
pybboxes==0.1.6
pyparsing==3.0.9
python-dateutil==2.8.2
pytz==2023.3
PyYAML==6.0
qudida==0.0.4
requests==2.28.2
safetensors==0.3.3
scikit-image==0.20.0
scikit-learn==1.2.2
scipy==1.10.1
seaborn==0.12.2
# segmentation-models-pytorch==0.3.3
six==1.16.0
smmap==5.0.0
thop==0.1.1.post2209072238
threadpoolctl==3.1.0
tifffile==2023.4.12
# timm==0.9.2
# torch==2.0.1+cu117
# torchvision==0.15.2+cu117
tqdm==4.65.0
typing_extensions==4.7.1
tzdata==2022.2
ultralytics==8.0.110
urllib3==1.26.16



##pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu117 

#####verify installation
####python
####import torch
####torch.cuda.is_available()

#Extra for DeepSort
torch
absl-py
git+https://github.com/THU-MIG/yolov10.git
deep-sort-realtime