search

lighttransport / c-libtorch

Experimental C binding for libtorch
MIT License
19 stars 1 forks source link

ROCm Request #2

Open Eudox67 opened 1 week ago

Eudox67 commented 1 week ago

I wish I new enough to help with this project. So intially, I want to express appreciation for the effort going into this project. I see that initial CUDA support has been added. Would like to request that ROCm also be added. One of the main issues with GPU cost is the restriction to nVidia GPU's. Hope others will join to help with this project.

syoyo commented 1 week ago

You can contribute!

You can also contribute ROCm support.