The TensorFlow GPU setup