pyflink.dataframe.get_gpu_infos#
- get_gpu_infos() List[pyflink.dataframe.gpu.GPUInfo][source]#
Get GPU information allocated to the current Python worker.
Reads the
_PYFLINK_GPU_INFOSenvironment variable set by the Java operator and returns a list ofGPUInfoobjects.- Returns
List of GPUInfo. Empty list if no GPU is allocated.