PyTorch version with support for GPU #147
-
|
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, |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment
-
|
Dear Jenkin, Yes you are correct, we are also working on an upcoming release of decode with a more recent pytorch version. In the meantime I would advise to clone the repository, change the Let me know if that works for you |
Beta Was this translation helpful? Give feedback.
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.yamlto 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