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 && \
|
nvidia-nccl-cu12==2.27.3 --force-reinstall --no-deps && \
|
||||||
# ✅ 补全依赖(必须)
|
# ✅ 补全依赖(必须)
|
||||||
python -m pip install --no-cache-dir \
|
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: 生成最小运行镜像 #
|
# Stage-1: 生成最小运行镜像 #
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue