Skip to content

Latest commit

 

History

History

tensorflow-landsat

Processing Landsat satellite images with GPUs

Open in Cloud Shell

📝 Tutorial: Processing Landsat satellite images with GPUs

Before you begin

Make sure you have followed the Dataflow setup instructions.

Building the Docker image

We use Cloud Build to build the container image for the workers and save it in Container Registry.

gcloud builds submit --config build.yaml

Running the Dataflow job with GPUs

We use Cloud Build to run the Dataflow job.

ℹ️ We launch the job using the worker image to make sure the job launches with the same Python version as the workers and all the dependencies installed.

export OUTPUT_PATH="gs://$BUCKET/samples/dataflow/landsat/output-images/"
export REGION="us-central1"
export GPU_TYPE="nvidia-tesla-t4"

gcloud builds submit \
    --config run.yaml \
    --substitutions _OUTPUT_PATH=$OUTPUT_PATH,_REGION=$REGION,_GPU_TYPE=$GPU_TYPE \
    --no-source