ultralytics
pytorch
torchvision
torchaudio
pytorch-cuda=12.4