Yan Bai
commited on
Commit
·
b125668
1
Parent(s):
72f5bfa
build
Browse files- Dockerfile +3 -2
Dockerfile
CHANGED
|
@@ -1,11 +1,12 @@
|
|
| 1 |
-
FROM
|
| 2 |
|
| 3 |
# 安装额外依赖（如果基础镜像已包含部分依赖，pip 会自动跳过）
|
| 4 |
RUN pip install --no-cache-dir \
|
| 5 |
fastapi \
|
| 6 |
uvicorn[standard] \
|
| 7 |
mbridge==0.13.1 \
|
| 8 |
-
termcolor
|
|
|
|
| 9 |
|
| 10 |
# 添加 Megatron-LM core_v0.12.2
|
| 11 |
RUN git clone -b core_v0.13.0rc4 --depth 1 https://github.com/NVIDIA/Megatron-LM.git /opt/Megatron-LM
|
|
|
|
| 1 |
+
FROM nvcr.io/nvidia/pytorch:25.09-py3
|
| 2 |
|
| 3 |
# 安装额外依赖（如果基础镜像已包含部分依赖，pip 会自动跳过）
|
| 4 |
RUN pip install --no-cache-dir \
|
| 5 |
fastapi \
|
| 6 |
uvicorn[standard] \
|
| 7 |
mbridge==0.13.1 \
|
| 8 |
+
termcolor \
|
| 9 |
+
transformers==4.57.0
|
| 10 |
|
| 11 |
# 添加 Megatron-LM core_v0.12.2
|
| 12 |
RUN git clone -b core_v0.13.0rc4 --depth 1 https://github.com/NVIDIA/Megatron-LM.git /opt/Megatron-LM
|