SkyPilot-3.yaml
· 572 B · YAML
Ham
resources:
accelerators: A100:8 # 8x NVIDIA A100 GPU
num_nodes: 1 # Number of VMs to launch
# Working directory (optional) containing the project codebase.
# Its contents are synced to ~/sky_workdir/ on the cluster.
workdir: ~/torch_examples
# Commands to be run before executing the job.
# Typical use: pip install -r requirements.txt, git clone, etc.
setup: |
pip install "torch<2.2" torchvision --index-url https://download.pytorch.org/whl/cu121
# Commands to run as a job.
# Typical use: launch the main program.
run: |
cd mnist
python main.py --epochs 1
| 1 | resources: |
| 2 | accelerators: A100:8 # 8x NVIDIA A100 GPU |
| 3 | |
| 4 | num_nodes: 1 # Number of VMs to launch |
| 5 | |
| 6 | # Working directory (optional) containing the project codebase. |
| 7 | # Its contents are synced to ~/sky_workdir/ on the cluster. |
| 8 | workdir: ~/torch_examples |
| 9 | |
| 10 | # Commands to be run before executing the job. |
| 11 | # Typical use: pip install -r requirements.txt, git clone, etc. |
| 12 | setup: | |
| 13 | pip install "torch<2.2" torchvision --index-url https://download.pytorch.org/whl/cu121 |
| 14 | |
| 15 | # Commands to run as a job. |
| 16 | # Typical use: launch the main program. |
| 17 | run: | |
| 18 | cd mnist |
| 19 | python main.py --epochs 1 |