yolo v5如何运用cuda去训练模型

yolo v5，请问如图所示，我的电脑明明是有cuda，为什么在pycharm里却显示不可用呢？那我应该如何使用cuda去训练模型呢？

第一张图只能证明你安装了nvidia驱动，但可能未安装cuda，
如果已经安装cuda，就要安装torch的cuda版本，而不是cpu版本。
安装cuda版的torch命令应该是

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116