...
There is only enough VRAM for nano and small models, not larger ones.
PyTorch version 2.4.0 1 is being used with CUDA 12.4, which has minimum driver requirements. Please ensure that a suitable driver is installed (there should be a high enough driver version for the RTX 2060, you can find them here if needed. (We are not going to use the old Jetson TX2i, but the RPi5 on the drone)
WARG desktop details: WARG Desktop
More detailed CUDA compatibility information: CUDA and PyTorch
...