mirror of
https://github.com/vale981/ray
synced 2025-03-09 04:46:38 -04:00
10 lines
196 B
Python
10 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)
|