Tensorflow with GPU