How to use the netron.cloud.AWSCluster function in netron

To help you get started, we’ve selected a few netron examples, based on popular ways it is used in public projects.

Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.

github yankov / netron / examples / create_gpu_cluster.py View on Github external
from netron.cloud import AWSCluster
import time

# This script creates and then destroys a 2-node GPU cluster in AWS.

cluster = AWSCluster()

# Create 2 GPU instances at max price of $0.1/hr.
cluster.create_spot_instances(max_spot_price = 0.20, instance_count = 2, bootstrap_script = "examples/bootstrap_worker.sh")
req = cluster.describe_spot_requests()

while "open" in req:
    time.sleep(1)
    print "Waiting to fullfill %d spot requests." % req["open"]
    req = cluster.describe_spot_requests()

print "All spot requests have been fulfilled!\nInstance running: %d" % cluster.live_instances_count()
print "Terminating ... "

# Clearing spot requests, just in case.
cluster.cancel_all_spot_requests()