This commit is contained in:
hailin 2025-06-30 15:31:40 +08:00
parent 0a15939326
commit 0ee2a26757
1 changed files with 2 additions and 2 deletions

View File

@ -28,9 +28,9 @@ RUN case "$CUDA_VERSION" in \
nvidia-nccl-cu12==2.27.3 --force-reinstall --no-deps && \
# ✅ 补全依赖(必须)
python -m pip install --no-cache-dir \
pydantic psutil pyzmq pynvml transformers==4.48.3 uvicorn fastapi IPython aiohttp setproctitle && \
pydantic psutil pyzmq pynvml transformers==4.48.3 uvicorn fastapi IPython aiohttp setproctitle
# ✅ 测试模块完整性
RUN python -c "import sglang, torch, pydantic, transformers"
#python -c "import sglang, torch, pydantic, transformers, sgl_kernel"
############################################################
# Stage-1: 生成最小运行镜像 #