Запуск обучения из командной строки:

yolo detect train data=data.yaml patience=100 epochs=1000 device=0 cache=ram batch=-1 save_period=100 project=fingerpints pretrained=False

Установка pycuda

export CPATH=$CPATH:/usr/local/cuda-11.8/targets/x86_64-linux/include
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-11.8/targets/x86_64-linux/lib
pip install pycuda