skypilot

1
name: gpu-task1
2

3
resources:
4
  # Optional; if left out, automatically pick the cheapest cloud.
5
  cloud: oci
6

7
  accelerators: A10:1  # 1x NVIDIA A10 GPU
8

9
  region: ap-seoul-1
10
  
11
  # zone: AP-SEOUL-1-AD-1
12
  
13
  # instance_type: VM.GPU.A10.1
14

15
  # image_id: skypilot:gpu-ubuntu-2004
16

17
# Working directory (optional) containing the project codebase.
18
# Its contents are synced to ~/sky_workdir/ on the cluster.
19
# For example, prepare the workdir by cloning:
20
# git clone https://github.com/pytorch/examples.git ~/torch_examples
21
workdir: ~/torch_examples
22

23
num_nodes: 1
24

25
# Typical use: pip install -r requirements.txt
26
# Invoked under the workdir (i.e., can use its files).
27
setup: |
28
  echo "*** Running setup. ***"
29
  pip install torch torchvision
30

31
# Typical use: make use of resources, such as running training.
32
# Invoked under the workdir (i.e., can use its files).
33
run: |
34
  echo "*** Running the task on OCI ***"
35
  cd mnist
36
  python main.py --epochs 1
37
  echo "The task is completed."
38