TuragaLab/DECODE

PyTorch version with support for GPU

tsuijenk opened this issue · 1 comments

Hello authors,

My understanding is that DECODE is built against pytorch 1.7.1. It does not work with the current version of pytorch 1.10. However, I need to be on version 1.10 to get support for my GPU.

Is there any way to get around this?

Thanks,
Jenkin

Dear Jenkin,

Yes you are correct, we are also working on an upcoming release of decode with a more recent pytorch version.
I already tried that out recently, and I think it was only minor tests that were failing.

In the meantime I would advise to clone the repository, change the environment.yaml to your pytorch/cudatoolkit needs and follow the developer instructions described here:
https://github.com/TuragaLab/DECODE/blob/master/DEVELOPER.md

Let me know if that works for you