num.cr icon indicating copy to clipboard operation
num.cr copied to clipboard
verified publisher icon crystal-data

CUDA support

Open watzon opened this issue 3 years ago • 2 comments

It would be really cool if this could support CUDA as a backend

watzon avatar Nov 28 '22 20:11 watzon

A CUDA backend would be way easier than the current OpenCL backend due to much more documentation and support. The only reason I haven’t done it is because I don’t have a CUDA compatible graphics card, although with prices coming down maybe I should snag one!

christopherzimmerman avatar Nov 28 '22 21:11 christopherzimmerman

Do it! I'd look into implementing it myself, but as many projects as I've tried to start in the past couple weeks I'm realizing I just don't have the bandwidth. I would, however, like to migrate cadmiumcr/glove to use num.cr for GPU support. It would be so much quicker to train the GLOVE models using the GPU.

watzon avatar Nov 29 '22 16:11 watzon