how to run code on GPU - Githubissues

Separius / BERT-keras

Keras implementation of BERT with pre-trained weights

GNU General Public License v3.0

813 stars 196 forks source link

how to run code on GPU #25

Closed niurouli closed 5 years ago

niurouli commented 5 years ago

i have installed tensorflow-gpu 1.7.0, but when i run this code on GPU, it got error, "ModuleNotFoundError: No module named 'tensorflow.contrib.tpu.python.tpu.keras_support'", how can i solve it?