yolov9 在 PC 上使用 GPU

Question

您好，我有一个问题或者更多的问题。我正在从事一个物体检测项目，我训练了我的 yolov9 模型，现在我想测试它。但每次我测试它时，我的网络摄像头的每秒帧数都很低。它使用我的 cpu 而不是 gpu 。

`Ultralytics YOLOv8.2.48 🚀 Python-3.12.4 torch-2.3.1+cpu CPU (AMD Ryzen 7 5800H with Radeon Graphics)
YOLOv9c summary (fused): 384 layers, 25321561 parameters, 0 gradients, 102.3 GFLOPs
`

当我测试 GPU 时，我得到：

`import torch
print(torch.__version__)

if torch.cuda.is_available():
  print(f"GPU Name: {torch.cuda.get_device_name(0)}")
  print(f"GPU Is Available: {torch.cuda.is_available()}")
else:
  print("GPU is not available")`

2.3.1+CPU GPU 不可用

有没有办法让它更快，而不是每 2 秒 1 帧 :/

我尝试安装cuda，我想使用GPU来更快地处理图像

Answer 1

检查 Nvidia 的网站，您的 GPU 是否兼容使用 CUDA。然后安装torch版本以使用GPU版本。在 PyTorch 网站中，您可以准确选择您希望其默认运行的 CUDA 版本和硬件（GPU/CPU）。

如果您的 GPU 不兼容并且您必须使用 CPU，那么您在这里无能为力。可以尝试使用 Google 的 Colab 进行一些解决方法，但我不确定这是否有效