Train a PyTorch model with a GPU

[mrocklin]

Maybe this doesn't make as much sense for this data, but I'd also like to have a model trained with PyTorch on a GPU machine if possible. Probably this is just ...

@task 
@coiled.function(vm_type=..., software=...)
def train_pytorch_model(...):
    do GPU stuff

@flow(...)
def keep_gpu_model_updated(...):
    ...

"How do I use this to train a deep learning model" is a pretty common question. It's valuable to show off that coiled.function makes using GPUs pretty easy.