torch==2.0.1+cu118
torchvision==0.15.2+cu118
torchaudio==2.0.2

numpy<2
opencv-python
Pillow
shapely
numba
tensorboard
gdown
pydantic
scikit-image