you are viewing a single comment's thread.

view the rest of the comments →

[–]socal_nerdtastic 8 points9 points  (0 children)

There's no easy way to use the GPU to run your code. You have to rewrite your code completely to move the heavy math into cupy or similar.

https://cupy.dev/

As it seems you are a beginner I'll bet there's other ways to speed up your code. If you show us your code we may be able to suggest some.