YOLO не видит GPU

Начал изучать nn. При обучении nn kernel крашится:

The Kernel crashed while executing code in the current cell or a previous cell. 

Please review the code in the cell(s) to identify a possible cause of the failure. 

Click here for more info. 

View Jupyter log for further details.

Мною было замечено, что YOLO использует ресурсы CPU для обучения нейронки, и, как я полагаю, именно из-за этого и возникает краш.

CUDNN и CUDA установлены, но yolo не видит видеокарту и не использует ее ресурсы. Как это можно исправить?