lightning/examples/app_multi_node/train_lt_script.py

13 lines
365 B
Python
Raw Normal View History

import lightning as L
from lightning.app.components import LightningTrainerScript
from lightning.app.utilities.packaging.cloud_compute import CloudCompute
# 8 GPUs: (2 nodes of 4 x v100)
app = L.LightningApp(
LightningTrainerScript(
"pl_boring_script.py",
num_nodes=2,
cloud_compute=CloudCompute("gpu-fast-multi"), # 4 x v100
),
)