42 lines
887 B
Python
42 lines
887 B
Python
# 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',
|
|
]
|