2022-11-04 17:41:59 +00:00
|
|
|
import lightning as L
|
2022-11-10 19:48:48 +00:00
|
|
|
from lightning.app.components import LightningTrainerScript
|
2022-07-29 14:44:52 +00:00
|
|
|
from lightning.app.utilities.packaging.cloud_compute import CloudCompute
|
|
|
|
|
2022-12-16 09:49:17 +00:00
|
|
|
# 8 GPUs: (2 nodes of 4 x v100)
|
2022-11-04 17:41:59 +00:00
|
|
|
app = L.LightningApp(
|
2022-11-10 19:48:48 +00:00
|
|
|
LightningTrainerScript(
|
2022-11-08 12:55:31 +00:00
|
|
|
"pl_boring_script.py",
|
2022-07-29 14:44:52 +00:00
|
|
|
num_nodes=2,
|
2022-12-16 09:49:17 +00:00
|
|
|
cloud_compute=CloudCompute("gpu-fast-multi"), # 4 x v100
|
2022-07-29 14:44:52 +00:00
|
|
|
),
|
|
|
|
)
|