Fix issues due to old numpy versions in dist_test and gcs_test

Previously numpy was installed with apt-get, leading to a version too
old to be compatible with the recently-added autograd dependency of
tensorflow. This change set fixes that.