skypilot

1
resources:
2
    cloud: aws
3
    accelerators: V100
4

5
num_nodes: 2
6

7
workdir: examples/ray_tune_examples
8

9
setup: |
10
    pip3 install --upgrade pip
11
    pip3 install ray[tune] pytorch-lightning==1.4.9 lightning-bolts torchvision
12

13
run: |
14
    if [ "${SKYPILOT_NODE_RANK}" == "0" ]; then
15
        python3 tune_ptl_example.py
16
    fi
17