skypilot

1
name: mnist-gan
2

3
resources:
4
  candidates:
5
  - {accelerators: T4}
6
  - {accelerators: V100}
7

8
workdir: ./examples/benchmark/lightning_gan
9

10
setup: |
11
  conda create -n pl python=3.8 -y
12
  conda activate pl
13

14
  # Install SkyCallback
15
  pip install "git+https://github.com/skypilot-org/skypilot.git#egg=sky-callback&subdirectory=sky/callbacks/"
16

17
  # User setup
18
  pip install "torchvision" "pytorch-lightning>=1.4" "torch>=1.6, <1.9"
19
  git clone https://github.com/Lightning-AI/tutorials.git
20
  cd tutorials
21
  git checkout e22e229921a97ea241277e19e0eaddedc35808cb
22

23
  # Apply the patch to enable SkyCallback
24
  git apply ../callback.patch
25

26
run: |
27
  conda activate pl
28
  cd tutorials/lightning_examples/basic-gan/
29
  python gan.py
30