torch>=2.0.1
torchvision>=0.15.2
opencv-python
tqdm
tensorboard