File tree Expand file tree Collapse file tree 2 files changed +39
-5
lines changed
Expand file tree Collapse file tree 2 files changed +39
-5
lines changed Original file line number Diff line number Diff line change @@ -26,14 +26,29 @@ This repository provides a unified evaluation framework for benchmarking **video
2626
2727## 🔧 Setup
2828
29+
30+ ### 1. Optional: Using Dockerfile for Environment Setup
31+ ``` bash
32+ docker build -t llava_vit_eval:25.09 .
33+ ```
34+ ### 2. Or Load Docker Image
2935``` bash
30- # Clone the repo
31- git clone git@github.com:FeilongTangmonash/Encoder_Eval.git
32- cd Encoder_Eval
36+ docker load -i /vlm/xiangan/docker_images/llava_vit_eval_tag_25.09.tar
37+ docker tag < image_id > llava_vit_eval:25.09
38+ ```
3339
34- # Install dependencies
35- pip install -r requirements.txt
40+ ### 3. Run
41+ ```
42+ # Run container with -w to set working directory directly to the mounted volume
43+ docker run -it --gpus all \
44+ --ipc host --net host --privileged --cap-add IPC_LOCK \
45+ --ulimit memlock=-1 --ulimit stack=67108864 --rm \
46+ -v "$(cd .. && pwd)":/workspace/LLaVA-ViT \
47+ -w /workspace/LLaVA-ViT/Encoder_Eval \
48+ --name "llava_vit_eval_container" \
49+ llava_vit_eval:25.09 /bin/bash
3650```
51+
3752## 🧱 code structure
3853
3954<pre >
Original file line number Diff line number Diff line change 1+ FROM pytorch/pytorch:2.7.0-cuda11.8-cudnn9-runtime
2+
3+ # Set up environment variables (optional, but often recommended)
4+ ENV DEBIAN_FRONTEND=noninteractive
5+
6+ # Install Python packages
7+ RUN pip install --no-cache-dir --extra-index-url https://pypi.nvidia.com --upgrade nvidia-dali-cuda110 \
8+ && pip install --no-cache-dir decord==0.6.0 \
9+ && pip install --no-cache-dir timm \
10+ && pip install --no-cache-dir transformers==4.52.4
11+
12+ # (Optional) Set working directory
13+ WORKDIR /workspace
14+
15+ # (Optional) Copy your code into the container
16+ # COPY . /workspace
17+
18+ # (Optional) Set entrypoint or CMD
19+ # CMD ["python", "your_script.py"]
You can’t perform that action at this time.
0 commit comments