search

tsudalab / ChemTS

Molecule Design using Monte Carlo Tree Search with Neural Rollout
156 stars 52 forks source link

how to set GPU for training? #7

Open xuzhang5788 opened 6 years ago

xuzhang5788 commented 6 years ago

I use your train_RNN.py, I got

Using TensorFlow backend. 2018-06-13 17:21:01.080746: I tensorflow/core/platform/cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA

But it show that I am using CPU instead of GPU. My keras is using GPU. Any advice? Many thanks

yangxiufengsia commented 6 years ago

In order to use GPU, you need to install a tensorflow that supports GPU. How did you installed tensorflow? did you follow the instructions here: https://www.tensorflow.org/install/install_sources

yangxiufengsia commented 6 years ago

@xuzhang5788