wehub-resource-sync
19 lines
396 B
Python
19 lines
396 B
Python
import ray
|
|
|
|
|
|
def main():
|
|
"""Requests placement of a GPU actor."""
|
|
|
|
@ray.remote(num_gpus=1, num_cpus=1)
|
|
class GPUActor:
|
|
def where_am_i(self):
|
|
assert len(ray.get_gpu_ids()) == 1
|
|
return "on-a-gpu-node"
|
|
|
|
GPUActor.options(name="gpu_actor", lifetime="detached").remote()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
ray.init("auto", namespace="gpu-test")
|
|
main()
|