mirror of
https://github.com/vale981/ray
synced 2025-03-10 13:26:39 -04:00
14 lines
256 B
Python
14 lines
256 B
Python
|
import time
|
||
|
|
from ray.cluster_utils import Cluster
|
||
|
|
|
||
|
|
cluster = Cluster()
|
||
|
|
|
||
|
|
cluster.add_node(num_cpus=16)
|
||
|
|
|
||
|
|
time.sleep(20)
|
||
|
|
print("Scaling up.")
|
||
|
|
cluster.add_node(num_cpus=16, num_gpus=1)
|
||
|
|
|
||
|
|
print("Scaled up. Waiting for 1000 seconds until done.")
|
||
|
|
time.sleep(1000)
|