PaddleX/zhch/utils/__init__.py

"""
工具模块包

包含CUDA环境检测、文件处理等实用工具函数
"""

from .check_cuda_env import (
    check_nvidia_environment,
)

from .cuda_utils import (
    monitor_gpu_memory,
    detect_available_gpus
)

from .file_utils import (
    get_image_files_from_dir,
    get_image_files_from_list,
    get_image_files_from_csv,
    split_files,
    create_temp_file_list
)

__all__ = [
    # CUDA环境检测
    'check_nvidia_environment',
    # CUDA工具
    'monitor_gpu_memory',
    'detect_available_gpus',
    # 文件工具
    'get_image_files_from_dir',
    'get_image_files_from_list',
    'get_image_files_from_csv',
    'split_files',
    'create_temp_file_list'
]

__version__ = "1.0.0"
__author__ = "zhch158"
__description__ = "PaddleX工具模块包，提供CUDA环境检测和文件处理功能"