Hi @hisisman3, if you installed PyTorch/torchvision from pip (or if installing a yolo package from pip pulled those in as dependencies), then the torch wheels from PyPi aren’t built with CUDA enabled. These PyTorch wheels were built with CUDA enabled:
There are also instructions in that post for building torchvision with CUDA. First you should pip3 uninstall torch torchvision to remove the versions that don’t have GPU acceleration.