search

lamoureux-lab / TorchProteinLibrary

PyTorch library of layers acting on protein representations
https://lamoureux-lab.github.io/TorchProteinLibrary/
MIT License
116 stars 23 forks source link

CPU-only version #32

Open leeyang opened 4 years ago

leeyang commented 4 years ago

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

lupoglaz commented 4 years ago

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

leeyang commented 3 years ago

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

lupoglaz commented 3 years ago

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.

gkementzidis commented 1 year ago

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