# Test multi-node, multi-GPU Ray SGD example.
- name: sgd_gpu
  cluster:
    app_config: sgd_gpu/sgd_gpu_app_config.yaml
    compute_template: sgd_gpu/sgd_gpu_compute.yaml

  run:
    timeout: 3000
    prepare: python wait_cluster.py 2 600
    script: python sgd_gpu/sgd_gpu_test.py --num-workers=2 --use-gpu --address=auto