У меня есть машина с Ubuntu с графическим процессором Nvidia. И я пытаюсь тренировать модель YOLO, используя для этого Jupyter Notebook.
Когда я запускаю команду top, я вижу процесс darknet, использующий 100% ЦП. Но когда я запускаю команду nvidia-smi, я не вижу запущенного процесса на графическом процессоре.
Я попробовал команды -gpus 0, чтобы заставить модель YOLO использовать GPU. Но до сих пор нет прогресса. Пожалуйста, дайте мне знать, если вы знаете какое-либо возможное решение.
Я действительно борюсь с этим, поскольку я новичок в GPU.