13 lines
360 B
Python
13 lines
360 B
Python
import lightning as L
|
|
from lightning.app.components import LightningTrainingComponent
|
|
from lightning.app.utilities.packaging.cloud_compute import CloudCompute
|
|
|
|
# Run over 2 nodes of 4 x V100
|
|
app = L.LightningApp(
|
|
LightningTrainingComponent(
|
|
"pl_boring_script.py",
|
|
num_nodes=2,
|
|
cloud_compute=CloudCompute("gpu-fast-multi"),
|
|
),
|
|
)
|