-2
$\begingroup$

My laptop has NVIDIA GeForce GTX1650 GPU. I want to utilize this GPU to run my Python script. Any help in the form of code would be really helpful. I mean tried researching this so much but I couldn't find.

Basically I am looping over all the frames of a video and at each frame I am using yolov8s-pose model to detect the keypoints. So since it is running on my CPU, it is very slow. So, that is the reason I need to know how to utilize my device's GPU to run it. Any help would be highly appreciated! Thank you.

Improve this question
$\endgroup$

1 Answer 1

Reset to default
0
$\begingroup$

That is going to depend on the library you use. In general, the term you should google is Cuda along with your GPU. In TensorFlow I have used things like this and I would recommend looking here and here.

Improve this answer
$\endgroup$
2
  • $\begingroup$ Awesome! I was able to use the GPU now. Thank you. $\endgroup$
    – Escanor6
    Commented Nov 8, 2023 at 12:57
  • $\begingroup$ This is a programming question. Programming questions are off-topic. Please, stop answering questions that are off-topic. If you're not familiar with the on-topic page, this is the perfect time to do it: ai.stackexchange.com/help/on-topic $\endgroup$
    – nbro
    Commented Nov 9, 2023 at 8:47

Not the answer you're looking for? Browse other questions tagged .