ray.获取_GPU_ID

ray.get_gpu_ids() → List[int] | List[str]

获取工作进程可用的GPU的ID。

此方法只能在任务或角色内部调用，而不能在驱动程序中调用。

如果在工作进程启动时设置了CUDA_VISIBLE_DEVICES环境变量，那么此方法返回的ID将是CUDA_VISIBLE_DEVICES中ID的子集。如果没有设置，ID将在范围[0, NUM_GPUS - 1]内，其中NUM_GPUS是节点拥有的GPU数量。

返回:

GPU ID 列表。