FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.11
ARG IMAGE=ollama
ARG IMAGE_UPPER=ollama
ARG BRANCH=dcu
RUN apt-get update && apt-get install -y curl && apt-get clean && \
    rm -rf /var/lib/apt/lists/*
COPY chenyh/$IMAGE/install.sh chenyh/$IMAGE/ollama  /root/
WORKDIR /root
RUN sh install.sh && rm install.sh
RUN pip install --index-url https://pypi.tuna.tsinghua.edu.cn/simple open-webui jupyter
RUN mkdir -p /opt/conda/bin/ && cp /usr/local/bin/jupyter /opt/conda/bin/