Skip to content

Commit 288a323

Browse files
authored
Merge pull request #3 from anxiangsir/dev_anxiang
docs(readme): update dockerfile
2 parents 04f3fe0 + 3382b39 commit 288a323

File tree

2 files changed

+39
-5
lines changed

2 files changed

+39
-5
lines changed

Encoder_Eval/README.md

Lines changed: 20 additions & 5 deletions
Original file line numberDiff line numberDiff 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>

Encoder_Eval/dockerfile

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
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"]

0 commit comments

Comments
 (0)