| 123456789101112131415161718192021222324252627282930313233343536373839404142 |
- """
- 工具模块包
- 包含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,
- collect_pid_files,
- )
- __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',
- 'collect_pid_files'
- ]
- __version__ = "1.0.0"
- __author__ = "zhch158"
- __description__ = "PaddleX工具模块包，提供CUDA环境检测和文件处理功能"
|
