1234567891011121314151617181920212223242526272829303132 |
- import os
- import socket
- import sys
- import time
- # trainer.py
- from collections import Counter
- import ray
- num_cpus = int(sys.argv[1])
- ray.init(address=os.environ["ip_head"])
- print("Nodes in the Ray cluster:")
- print(ray.nodes())
- @ray.remote
- def f():
- time.sleep(1)
- return socket.gethostbyname(socket.gethostname())
- # The following takes one second (assuming that
- # ray was able to access all of the allocated nodes).
- for i in range(60):
- start = time.time()
- ip_addresses = ray.get([f.remote() for _ in range(num_cpus)])
- print(Counter(ip_addresses))
- end = time.time()
- print(end - start)
|