618f45bab1
* Upgrade habana docker version to 1.18.0 Signed-off-by: lvliang-intel <liang1.lv@intel.com> * fix issues Signed-off-by: lvliang-intel <liang1.lv@intel.com> * fix ci issue Signed-off-by: lvliang-intel <liang1.lv@intel.com> --------- Signed-off-by: lvliang-intel <liang1.lv@intel.com>
33 lines
778 B
Docker
33 lines
778 B
Docker
# Copyright (C) 2024 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
# HABANA environment
|
|
FROM vault.habana.ai/gaudi-docker/1.18.0/ubuntu22.04/habanalabs/pytorch-installer-2.4.0 AS hpu
|
|
|
|
ENV LANG=en_US.UTF-8
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends --fix-missing \
|
|
git-lfs \
|
|
libgl1-mesa-glx \
|
|
libjemalloc-dev
|
|
|
|
RUN useradd -m -s /bin/bash user && \
|
|
mkdir -p /home/user && \
|
|
chown -R user /home/user/
|
|
|
|
USER user
|
|
|
|
RUN git lfs install
|
|
|
|
COPY comps /home/user/comps
|
|
|
|
RUN cd /home/user/comps/lvms/llama-vision/ && \
|
|
pip install --no-cache-dir -r requirements.txt && \
|
|
pip install --no-cache-dir --upgrade Pillow
|
|
|
|
ENV PYTHONPATH=/root:/home/user
|
|
|
|
WORKDIR /home/user/comps/lvms/llama-vision/
|
|
|
|
ENTRYPOINT ["python", "lvm.py"]
|