cf69b916af
remove "return state" for functions which will be executed in vmap; recover randkey as args in mutation methods
6 lines
132 B
Python
6 lines
132 B
Python
import ray
|
|
ray.init(num_gpus=2)
|
|
|
|
available_resources = ray.available_resources()
|
|
print("Available resources:", available_resources)
|