Skip to main content
Ctrl+K
PyFlink 1.20+vvr.11.7.dev0 documentation - Home PyFlink 1.20+vvr.11.7.dev0 documentation - Home
  • API Reference
  • Examples
  • API Reference
  • Examples

Section Navigation

  • PyFlink Table
  • PyFlink DataStream
  • PyFlink DataFrame
    • DataFrame
    • DataFrame Creation
    • Input/Output
    • SQL
    • DataType
    • User Defined Functions
    • Configuration
    • GPU Support
    • AI / LLM
  • PyFlink Common
  • API Reference
  • PyFlink DataFrame
  • GPU Support
  • pyflink.dataframe.gpu.get_gpu_infos

pyflink.dataframe.gpu.get_gpu_infos#

get_gpu_infos() → List[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.gpu.GPUInfo

next

AI / LLM

On this page
  • get_gpu_infos()

This Page

  • Show Source

Created using Sphinx 7.4.7.

Built with the PyData Sphinx Theme 0.16.1.