search

LPDI-EPFL / masif

MaSIF- Molecular surface interaction fingerprints. Geometric deep learning to decipher patterns in molecular surfaces.
Apache License 2.0
582 stars 154 forks source link

The provide docker file do not support tensorflow GPU #16

Closed zshyang closed 4 years ago

zshyang commented 4 years ago

Hi, Thanks for sharing this great work. But the docker file you provided seems do not have the GPU included. Do you know how could I include GPU and TensorFlow-GPU into the docker image? Best regards!

pablogainza commented 4 years ago

Hi! So I'm not sure how to include these - I've never run it. Did you try this tutorial? https://www.tensorflow.org/install/docker

zshyang commented 4 years ago

Thanks for your reply. I did not try the tutorial. And It seems that the docker you provided is only for data processing. Once the data processing is done. I could use some virtual environment with TensorFlow installed. Do I understand correctly?

pablogainza commented 4 years ago

So I think that should work yes.

On Thu, Sep 24, 2020 at 7:51 PM zshyang notifications@github.com wrote:

Thanks for your reply. I did not try the tutorial. And It seems that the docker you provided is only for data processing. Once the data processing is done. I could use some virtual environment with TensorFlow installed. Do I understand correctly?

— You are receiving this because you modified the open/close state. Reply to this email directly, view it on GitHub https://github.com/LPDI-EPFL/masif/issues/16#issuecomment-698494023, or unsubscribe https://github.com/notifications/unsubscribe-auth/AB7GNQUDUXL4EGHWZRVYM3DSHOBLTANCNFSM4RWR5OZQ .