TorchProteinLibrary icon indicating copy to clipboard operation
TorchProteinLibrary copied to clipboard
verified publisher icon lupoglaz

CPU-only version

Open leeyang opened this issue 5 years ago • 4 comments

Very nice work! But is it possible to install TorchProteinLibrary on a CPU only (without cuda) environment?

leeyang avatar Jul 08 '20 04:07 leeyang

Yes, it is possible, but some layers are hopelessly slow on cpu. I'll add an option to compile cpu-only version.

lupoglaz avatar Jul 13 '20 01:07 lupoglaz

Hi! Any progress? I find your method is much faster than another methods even though I implemented it simply using raw Pytorch autograd.

leeyang avatar Oct 21 '20 13:10 leeyang

I am working on it, but it's a huge pita: I have cuda kernels and cpu side in the same extension modules. Might take a week/two more.

lupoglaz avatar Oct 24 '20 03:10 lupoglaz

Hi, are there any updates on the CPU-only version? Thanks.

gkementzidis avatar Jul 24 '23 10:07 gkementzidis