# GPU Utilities Package
"""
외부 GPU 연결 및 관리 유틸리티

사용 예시:
    # GPU 상태 확인
    from app.AI_modules.gpu_utils import check_gpu_status
    check_gpu_status()

    # 원격 GPU 서버 연결
    from app.AI_modules.gpu_utils import RemoteGPUClient
    client = RemoteGPUClient(host='192.168.1.100', port=22)
    client.connect()

    # TensorFlow GPU 설정
    from app.AI_modules.gpu_utils import setup_tensorflow_gpu
    setup_tensorflow_gpu(memory_limit=4096)
"""

from .gpu_config import (
    check_gpu_status,
    get_gpu_info,
    setup_tensorflow_gpu,
    setup_pytorch_gpu,
    limit_gpu_memory,
)

from .remote_gpu import (
    RemoteGPUClient,
    SSHGPURunner,
)

__all__ = [
    'check_gpu_status',
    'get_gpu_info',
    'setup_tensorflow_gpu',
    'setup_pytorch_gpu',
    'limit_gpu_memory',
    'RemoteGPUClient',
    'SSHGPURunner',
]
