run docker as specific user
Browse files- Dockerfile +12 -3
- startup.sh +7 -3
Dockerfile
CHANGED
|
@@ -16,10 +16,19 @@ RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86
|
|
| 16 |
|
| 17 |
# Put conda in path so we can use conda activate
|
| 18 |
ENV PATH=$CONDA_DIR/bin:$PATH
|
| 19 |
-
|
| 20 |
RUN conda install --yes pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
|
| 21 |
|
| 22 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
|
| 24 |
RUN git clone https://github.com/TRI-ML/vlm-evaluation.git
|
| 25 |
|
|
@@ -27,4 +36,4 @@ RUN cd vlm-evaluation && pip install -e .
|
|
| 27 |
|
| 28 |
RUN pip install -e .
|
| 29 |
|
| 30 |
-
CMD ["sh", "/
|
|
|
|
| 16 |
|
| 17 |
# Put conda in path so we can use conda activate
|
| 18 |
ENV PATH=$CONDA_DIR/bin:$PATH
|
|
|
|
| 19 |
RUN conda install --yes pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
|
| 20 |
|
| 21 |
+
# User setup
|
| 22 |
+
RUN useradd -m -u 1000 user
|
| 23 |
+
|
| 24 |
+
USER user
|
| 25 |
+
|
| 26 |
+
ENV HOME=/home/user \
|
| 27 |
+
PATH=/home/user/.local/bin:$PATH
|
| 28 |
+
|
| 29 |
+
WORKDIR $HOME/app
|
| 30 |
+
|
| 31 |
+
COPY --chown=user . $HOME/app
|
| 32 |
|
| 33 |
RUN git clone https://github.com/TRI-ML/vlm-evaluation.git
|
| 34 |
|
|
|
|
| 36 |
|
| 37 |
RUN pip install -e .
|
| 38 |
|
| 39 |
+
CMD ["sh", "/app/startup.sh"]
|
startup.sh
CHANGED
|
@@ -8,9 +8,13 @@ P2=$!
|
|
| 8 |
sleep 30
|
| 9 |
echo "PWD:"
|
| 10 |
pwd
|
| 11 |
-
echo
|
| 12 |
-
|
| 13 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
echo $HF_TOKEN > .hf_token
|
| 15 |
python -m interactive_demo --port 40000 --model_id prism-dinosiglip+7b
|
| 16 |
python -m interactive_demo --port 40001 --model_family llava-v15 --model_id llava-v1.5-7b --model_dir liuhaotian/llava-v1.5-7b
|
|
|
|
| 8 |
sleep 30
|
| 9 |
echo "PWD:"
|
| 10 |
pwd
|
| 11 |
+
echo "LS:"
|
| 12 |
+
ls -als
|
| 13 |
+
|
| 14 |
+
echo "TEST SECRET: $TEST_SECRET"
|
| 15 |
+
|
| 16 |
+
echo "ENV: $(env)"
|
| 17 |
+
|
| 18 |
echo $HF_TOKEN > .hf_token
|
| 19 |
python -m interactive_demo --port 40000 --model_id prism-dinosiglip+7b
|
| 20 |
python -m interactive_demo --port 40001 --model_family llava-v15 --model_id llava-v1.5-7b --model_dir liuhaotian/llava-v1.5-7b
|