search

topazape / LSTM_Chem

Implementation of the paper - Generative Recurrent Networks for De Novo Drug Design.
The Unlicense
116 stars 55 forks source link

GPU for Tensorflow #28

Closed AyushM07 closed 2 years ago

AyushM07 commented 2 years ago

Hello, I was trying to run the Fine tuning notebook with a different dataset for my model and it is taking too long, I was wondering how to use the GPU version of Tensorflow to run the code? It would be great if you could help. Thank you!