colombus-dev/mlprofiler

GitHub: colombus-dev/mlprofiler

mlprofiler - 机器学习管道性能分析器

Stars: 1 | Forks: 0

# ML 分析器 通过分析源代码并对管道步骤进行分类来分析 ML 管道。 ## 要求 - Docker ≥ 28.5.1 - Docker Compose ≥ 2.40.0 - _(高级)_ 需要Docker Model Runner (https://docs.docker.com/ai/model-runner/get-started/) ≥ 1.1.37 以使用 LLM 分析器功能 ## 快速开始 ``` cd docker cp .env.sample .env # edit .env if needed, the defaults work for basic setup docker compose --env-file .env up --build ``` ## 高级设置(LLM 分析器) LLM 分析器功能需要 Docker Model Runner(DMR)。设置因平台而异。 ### Mac DMR 无需额外安装。DMR 通过 Metal(Apple Silicon)本地使用 llama.cpp。 ``` docker compose -f docker-compose.yml -f docker-compose.mac.yml --env-file .env up --build ``` ### Linux DMR [设置 vLLM 推理后端](https://docs.docker.com/ai/model-runner/inference-engines/#setting-up-vllm): ``` docker model install-runner --backend vllm --gpu cuda ``` [配置模型](https://docs.docker.com/ai/model-runner/inference-engines/#vllm-configuration): ``` docker model configure --hf_overrides '{ "max_model_len": 8192, "max_num_seqs": 10, "gpu_memory_utilization": 0.8, "enforce_eager": true }' hf.co/Qwen/Qwen2.5-Coder-7B-Instruct-AWQ ``` ``` docker compose -f docker-compose.yml -f docker-compose.linux.dmr.yml --env-file .env up --build ``` ### Linux vLLM(独立) 在 `.env` 中设置 `VLLM_FORCE=1`,然后: ``` docker compose -f docker-compose.yml -f docker-compose.linux.dmr.yml -f docker-compose.linux.vllm.yml --env-file .env up --build ``` ## 监控(可选) | 栈 | 命令 | |---|---| | Langfuse ([文档](https://langfuse.com/docs)) | `docker compose -f docker-compose-monitor-langfuse.yml up` | | Grafana/Prometheus ([文档](https://github.com/vllm-project/vllm/tree/main/examples/online_serving/prometheus_grafana)) | `docker compose -f docker-compose-monitor-grafana.yml up` | ## 开发 安装 pre-commit 钩子: ``` uv run --with pre-commit pre-commit install ``` ## 故障排除 **连接到 Docker 套接字时出现“权限被拒绝”** ``` sudo usermod -a -G docker $USER && newgrp docker ``` 然后重启。 **构建期间 `uv sync` 失败** 重新生成锁文件: ``` uv lock ``` **无法选择设备驱动程序 "nvidia"** 安装 [NVIDIA 容器工具包]() 并 [配置 Docker 以使用 NVIDIA 运行时](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html#configuring-docker).
标签:版权保护, 请求拦截, 逆向工具