TensorFlow 1.4, tensorflow-gpu package on pip.
As a side note, I’m not sure that batch_size=242 is correct in the Keras implementation, I’m a bit confused because it seems that PyTorch and Keras have a different semantic for this term. I edited my code, now it should work.