mirror of
https://github.com/vale981/ray
synced 2025-03-08 19:41:38 -05:00
9 lines
196 B
Python
9 lines
196 B
Python
import ray
|
|
import time
|
|
|
|
ray.init()
|
|
|
|
while ray.cluster_resources().get("GPU", 0) != 2:
|
|
print("Waiting for GPUs {}/2".format(ray.cluster_resources().get(
|
|
"GPU", 400)))
|
|
time.sleep(5)
|