0
$\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.

$\endgroup$
2
  • $\begingroup$ Please clarify your specific problem or provide additional details to highlight exactly what you need. As it's currently written, it's hard to tell exactly what you're asking. $\endgroup$
    – Community Bot
    Commented Nov 6, 2023 at 23:18
  • $\begingroup$ Depends on your software framework (TensorFlow, PyTorch etc.) there are different steps to setup. Take TensorFlow as example, please read this official doc as a start. $\endgroup$
    – lpounng
    Commented Nov 7, 2023 at 9:29

1 Answer 1

1
$\begingroup$

While I am not familiar with the full extent, it seems that Modular's Mojo uses the GPU to greatly boost performance. You can use native python code, or use their syntax to enhance the code you're writing.

I am just getting started in the ML field, so I am still working to learn and understand the tools available. Mojo is just one that I've been keeping an eye on lately.

Hope this helps!

$\endgroup$

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service and acknowledge you have read our privacy policy.

Not the answer you're looking for? Browse other questions tagged or ask your own question.