Machine Learning on Coiled
Back to modules
Course progress0%
article
GPU development loop
Move a local experiment onto temporary GPU hardware.
GPU Development Loop
Coiled can make remote GPUs feel like part of the normal Python loop. The trick is to keep experiments interactive while making artifacts portable.
Common loop
- Start a GPU notebook for exploration.
- Move stable training code into a script or function.
- Log parameters and metrics.
- Return CPU-compatible artifacts when calling from a local CPU machine.
Function sketch
import coiled
@coiled.function(vm_type="g5.xlarge", keepalive="20 minutes")
def train(params):
model = fit_model(params)
return model.to("cpu")
Handoff habit
Keep checkpoints, metrics, and model cards somewhere durable. The GPU VM should be disposable.
1
GPU development loop
GPU development