diff --git a/packaging/docker/DockerfileTDgpt b/packaging/docker/DockerfileTDgpt index 52e4a2c882..a381b32ab4 100644 --- a/packaging/docker/DockerfileTDgpt +++ b/packaging/docker/DockerfileTDgpt @@ -3,9 +3,15 @@ WORKDIR /apps ENV DEBIAN_FRONTEND=noninteractive ARG pkgFile ARG dirName -ARG nas_ip -ADD http://${nas_ip}/data/nas/TDengine/anode/taos.pth /apps -ADD http://${nas_ip}/data/nas/TDengine/anode/taos_ts_server.py /apps +ARG nasIp +ADD ${pkgFile} \ + ${dirName} \ + http://${nasIp}/data/nas/TDengine/anode/taos.pth \ + http://${nasIp}/data/nas/TDengine/anode/taos_ts_server.py \ + http://${nasIp}/data/nas/TDengine/anode/fc_australia.sql \ + http://${nasIp}/data/nas/TDengine/anode/timer-moe.tar.gz \ + /apps/ + ADD ${pkgFile} /apps/ RUN cd ${dirName}/ && /bin/bash install.sh -e no && cd .. && rm -rf ${dirName} COPY entrypoint.sh /usr/local/bin/entrypoint.sh diff --git a/packaging/docker/DockerfileTDgpt.base b/packaging/docker/DockerfileTDgpt.base index be9868ec47..5fc3c29190 100644 --- a/packaging/docker/DockerfileTDgpt.base +++ b/packaging/docker/DockerfileTDgpt.base @@ -10,6 +10,7 @@ RUN apt-get -o Acquire::Check-Valid-Until=false -o Acquire::Check-Date=false upd python3.10 -m venv /var/lib/taos/taosanode/venv && \ pip install --upgrade pip && \ pip install --no-cache-dir --ignore-installed blinker && \ + pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu && \ pip install --no-cache-dir \ numpy==1.26.4 \ pandas==1.5.0 \ @@ -21,9 +22,9 @@ RUN apt-get -o Acquire::Check-Valid-Until=false -o Acquire::Check-Date=false upd flask \ matplotlib \ uwsgi \ - transformers \ + transformers==4.40.0 \ + accelerate \ -i https://pypi.tuna.tsinghua.edu.cn/simple && \ - pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu && \ pip install --upgrade keras -i https://pypi.tuna.tsinghua.edu.cn/simple FROM python:3.10-slim