Tensorflow incompatable

Hi @wallarm 

I have tried tensorflow 1.4.0 and 1.5.0, however while running the train.py file I am getting this issue, 
` Module not found error: No module named '_pywrap_tensorflow_internal' `