This commit is contained in:
parent
0a15939326
commit
0ee2a26757
|
|
@ -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: 生成最小运行镜像 #
|
||||
|
|
|
|||
Loading…
Reference in New Issue