alessandro trinca tornidor commited on
Commit
4913772
·
1 Parent(s): 076e9b3

[refactor] revert to not-modified repository with a working Dockerfile and entrypoint.sh

Browse files
Files changed (2) hide show
  1. Dockerfile +48 -6
  2. scripts/entrypoint.sh +40 -1
Dockerfile CHANGED
@@ -1,19 +1,61 @@
1
- FROM nvidia/cuda:12.0.0-runtime-ubuntu22.04
2
  FROM nvcr.io/nvidia/pytorch:24.01-py3
 
 
 
3
  ARG DEBIAN_FRONTEND=noninteractive
 
4
 
5
  ENV PYTHONUNBUFFERED=1
 
 
 
6
 
7
- RUN apt-get update && apt-get install --no-install-recommends -y \
 
 
 
 
 
 
8
  build-essential \
9
  python3.11 \
10
  python3-pip \
 
 
11
  git \
12
  ffmpeg \
 
13
  && apt-get clean && rm -rf /var/lib/apt/lists/*
14
- WORKDIR /workspace
15
 
16
- COPY scripts/entrypoint.sh /workspace/entrypoint.sh
17
- RUN chmod +x /workspace/entrypoint.sh
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
18
 
19
- CMD ["/workspace/entrypoint.sh"]
 
 
 
 
 
 
 
 
 
 
 
1
  FROM nvcr.io/nvidia/pytorch:24.01-py3
2
+
3
+ LABEL authors="[email protected]"
4
+
5
  ARG DEBIAN_FRONTEND=noninteractive
6
+ ARG WORKDIR="/var/task"
7
 
8
  ENV PYTHONUNBUFFERED=1
9
+ ENV PYTHONPATH=${WORKDIR}:${WORKDIR}/venv:${PYTHONPATH}
10
+ ENV PATH=${WORKDIR}/venv/bin:$PATH
11
+ ENV XDG_CACHE_HOME=/data
12
 
13
+ WORKDIR ${WORKDIR}
14
+ COPY . ${WORKDIR}/
15
+ RUN ls ${WORKDIR}/
16
+ RUN mkdir -p ${XDG_CACHE_HOME}/.cache
17
+ RUN chmod 770 ${XDG_CACHE_HOME}/.cache
18
+
19
+ RUN apt update && apt upgrade -y && apt install --no-install-recommends -y \
20
  build-essential \
21
  python3.11 \
22
  python3-pip \
23
+ python3-dev \
24
+ python3-venv \
25
  git \
26
  ffmpeg \
27
+ curl \
28
  && apt-get clean && rm -rf /var/lib/apt/lists/*
 
29
 
30
+ RUN which python3
31
+ RUN python3 --version
32
+ RUN python3 -m venv venv
33
+ RUN source ${WORKDIR}/venv/bin/activate python -m pip install pip --upgrade && python -m pip install -r ${WORKDIR}/requirements.txt
34
+ RUN source ${WORKDIR}/venv/bin/activate && which python && python --version
35
+ RUN chmod +x ${WORKDIR}/scripts/entrypoint.sh
36
+ RUN curl -o /tmp/frpc_linux_amd64_v0.2 https://cdn-media.huggingface.co/frpc-gradio-0.2/frpc_linux_amd64
37
+ RUN ls -l /tmp/frpc_linux_amd64_v0.2
38
+ RUN cp /tmp/frpc_linux_amd64_v0.2 ${WORKDIR}/venv/lib/python*/site-packages/gradio
39
+ RUN ls -l ${WORKDIR}/venv/lib/python*/site-packages/gradio
40
+ RUN ls -l ${WORKDIR}/venv/bin
41
+ RUN bash --version
42
+ RUN chmod 770 ${WORKDIR}/flagged/
43
+ RUN chmod 770 ${WORKDIR}/flagged/* || true
44
+ RUN ls -ld ${WORKDIR}/flagged/
45
+ RUN ls -ld ${WORKDIR}/flagged/* || echo "folders ${WORKDIR}/flagged/* not found"
46
+ RUN ls -l ${WORKDIR}
47
+ RUN ls -l ${WORKDIR}/scripts/
48
+ RUN ls -l ${WORKDIR}/scripts/entrypoint.sh
49
+
50
+ EXPOSE 7860
51
 
52
+ CMD ["/var/task/scripts/entrypoint.sh"]
53
+ # CMD [
54
+ # "/var/task/scripts/entrypoint.sh",
55
+ # "/var/task/venv/bin/uvicorn", "app:lisa_app",
56
+ # "--host", "0.0.0.0",
57
+ # "--port", "7860",
58
+ # "--version='xinlai/LISA-13B-llama2-v1-explanatory'",
59
+ # "--precision='fp16'",
60
+ # "--load_in_4bit"
61
+ # ]
scripts/entrypoint.sh CHANGED
@@ -1,10 +1,49 @@
1
  #!/usr/bin/env bash
2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  free -m
4
- which python3
5
  which nvcc
6
  nvcc -V
7
  which nvidia-smi
8
  nvidia-smi
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
 
10
  exit 0
 
1
  #!/usr/bin/env bash
2
 
3
+ WORKDIR="/var/task"
4
+ XDG_CACHE_HOME="/data"
5
+ MPLCONFIGDIR=${XDG_CACHE_HOME}/.cache/matplotlib
6
+ TRANSFORMERS_CACHE=${XDG_CACHE_HOME}/.cache/transformers
7
+
8
+ ls -ld ${XDG_CACHE_HOME}/
9
+ ls -l ${XDG_CACHE_HOME}/
10
+
11
+ mkdir -p ${XDG_CACHE_HOME}/.cache
12
+ chmod 770 -R ${XDG_CACHE_HOME}/.cache
13
+ mkdir -p ${MPLCONFIGDIR}
14
+ mkdir -p ${TRANSFORMERS_CACHE}
15
+
16
+ ls -ld ${XDG_CACHE_HOME}/
17
+ ls -l ${XDG_CACHE_HOME}/
18
+ export WORKDIR
19
+ export XDG_CACHE_HOME
20
+ export MPLCONFIGDIR
21
+ export TRANSFORMERS_CACHE
22
+
23
+ source ${WORKDIR}/venv/bin/activate
24
+
25
+ which python
26
+ python --version
27
+
28
  free -m
 
29
  which nvcc
30
  nvcc -V
31
  which nvidia-smi
32
  nvidia-smi
33
+ pip list
34
+
35
+ df -h / /data /home /var/task
36
+
37
+ echo "WORKDIR - /var/task::"
38
+ ls -l ${WORKDIR}
39
+
40
+ echo "XDG_CACHE_HOME - /data"
41
+ find ${XDG_CACHE_HOME}
42
+
43
+ CUDA_VISIBLE_DEVICES=$(nvidia-smi --query-gpu=memory.free,index --format=csv,nounits,noheader | sort -nr | head -1 | awk '{ print $NF }')
44
+ echo "calculated CUDA_VISIBLE_DEVICES env variable: ${CUDA_VISIBLE_DEVICES}."
45
+ export CUDA_VISIBLE_DEVICES
46
+
47
+ python ${WORKDIR}/app.py --version='xinlai/LISA-13B-llama2-v1-explanatory' --precision='fp16' --load_in_4bit
48
 
49
  exit 0