Skip to content

Instantly share code, notes, and snippets.

@rikturr

rikturr/saturn-gpu-cluster.py

Created October 13, 2020 19:53
Show Gist options
  • Save rikturr/8b5bcead7c4cbd6674c48b28fbfa2c77 to your computer and use it in GitHub Desktop.
Save rikturr/8b5bcead7c4cbd6674c48b28fbfa2c77 to your computer and use it in GitHub Desktop.
Download ZIP
saturn-gpu-cluster
Raw
saturn-gpu-cluster.py
from dask.distributed import Client
from dask_saturn import SaturnCluster
cluster = SaturnCluster(
n_workers=3,
scheduler_size='medium',
worker_size='g4dnxlarge'
)
client = Client(cluster)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment