Skip to main content

工具包,包含 Redis 安装、基于 Redis 的GPU 任务队列管理等功能

Project description

GPU 自动选择、排队

  • gpu_queue(config: Union[DictConfig, dict] = {}):设置处理器
    支持自动选择和手动选择、排队和不排队
    
    Args:
        config (dict, optional): Defaults to {}.
            queuing: 是否排队,依赖 Redis,默认为 True
            visible_devices: 可见的 GPU 序号,用逗号分隔,默认为 -1, 表示全部可见
            memo: 备注
    Returns:
        Union[DictConfig, dict]: config
    

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

hi_spike-0.1.1.tar.gz (10.8 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page