Based on the Medium post on https://medium.com/fullstackai/how-to-train-an-object-detector-with-your-own-coco-dataset-in-pytorch-319e7090da5
Python 3.8 Pytorch: pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu115 CUDA 11.6