FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-py3.10-cuda12.1-ubuntu22.04-devel as base
ARG IMAGE=ollama
ARG IMAGE_UPPER=ollama
ARG BRANCH=gpu
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 chenyh/$IMAGE/node-v20.16.0-linux-x64.tar.xz  /root/
WORKDIR /root
RUN sh install.sh && rm install.sh
RUN /opt/conda/bin/conda create --name open-webui-env python=3.11 open-webui -y && \
    /opt/conda/bin/conda clean --all -f && pip install open-webui



#RUN apt-get purge -y nodejs npm --auto-remove
#RUN tar xf node-v20.16.0-linux-x64.tar.xz  -C /usr/local/
#RUN ln -sfv /usr/local/node-v20.16.0-linux-x64/bin/node /usr/bin/node && \
#    ln -sfv /usr/local/node-v20.16.0-linux-x64/bin/node /usr/local/bin/node && \
#    ln -sfv /usr/local/node-v20.16.0-linux-x64/bin/npm /usr/bin/npm && \
#    ln -sfv /usr/local/node-v20.16.0-linux-x64/bin/npm /usr/local/bin/npm
#
#
#RUN git clone http://developer.hpccube.com/codes/chenpangpang/open-webui.git
#RUN cd open-webui/ && cp -RPp .env.example .env && \
#    npm config set registry https://registry.npmmirror.com && \
#    npm install --verbose && npm run build && cd ./backend && pip install -r requirements.txt