使用GPU在Jupyter Notebook上运行进程

Question

我有一台配备Nvidia GPU的Ubuntu机器。我正在尝试使用Jupyter Notebook在其上训练YOLO模型。 当我运行top命令时，我看到使用100％CPU的Darknet进程。但是当我运行nvidia-smi命令时，我看到没有进程在GPU上运行。

我尝试了-gpus 0命令来强制YOLO模型使用GPU。但是仍然没有进展。如果您知道任何可能的解决方案，请告诉我。

由于我是GPU的新手，所以我真的很努力。