Ctrl+K
Logo image Logo image

Site Navigation

  • API Reference
  • Examples

Site Navigation

  • API Reference
  • Examples

Section Navigation

  • PyFlink Table
  • PyFlink DataFrame
    • DataFrame
    • DataFrame Creation
    • Input/Output
    • SQL
    • Data Types
    • User Defined Functions
    • Configuration
    • Catalog
    • GPU Support
    • AI / LLM
  • PyFlink Multimodal
  • PyFlink DataStream
  • PyFlink Common

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_INFOS environment variable set by the Java operator and returns a list of GPUInfo objects.

Returns

List of GPUInfo. Empty list if no GPU is allocated.

previous

pyflink.dataframe.GPUInfo

next

AI / LLM

Show Source

Created using Sphinx 4.5.0.