21 lines
520 B
Plaintext
21 lines
520 B
Plaintext
FROM harbor.4pd.io/mic-llm-x/combricon-mlu370x4-base:v0.2.0-tgiv1.4.3-btv0.6.0-pt2.1-x86_64-ubuntu22.04-py310
|
|
|
|
|
|
RUN if [ ! -d "/torch/venv3/pytorch" ]; then echo "虚拟环境路径缺失！"; exit 1; fi
|
|
|
|
|
|
WORKDIR /workspace/
|
|
COPY ./model_test_caltech_http_mlu370.py /workspace/
|
|
COPY ./microsoft_beit_base_patch16_224_pt22k_ft22k /model
|
|
|
|
|
|
# 安装transformers 4.46.3
|
|
RUN pip install transformers==4.46.3 uvicorn\[standard\] fastapi flask==3.1.1
|
|
|
|
EXPOSE 80
|
|
|
|
ENTRYPOINT ["python3", "model_test_caltech_http_mlu370.py"]
|
|
|
|
|
|
|