mirror of
https://github.com/Fancy-MLLM/R1-Onevision.git
synced 2026-01-13 23:07:18 +08:00
43 lines
1.1 KiB
Docker
Executable File
43 lines
1.1 KiB
Docker
Executable File
# Use the specified base image
|
|
FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-devel
|
|
|
|
# Set working directory
|
|
WORKDIR /workspace
|
|
|
|
# Install system dependencies
|
|
RUN apt-get update && apt-get install -y \
|
|
git \
|
|
wget \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Install flash-attn using pre-built wheel
|
|
RUN pip install --no-cache-dir \
|
|
"https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.5cxx11abiFALSE-cp311-cp311-linux_x86_64.whl" || \
|
|
pip install flash-attn --no-build-isolation
|
|
|
|
# Install additional required packages
|
|
RUN pip install \
|
|
wandb==0.18.3 \
|
|
tensorboardx \
|
|
qwen_vl_utils \
|
|
torchvision \
|
|
git+https://github.com/huggingface/transformers.git
|
|
|
|
# Copy local open-r1-multimodal repository
|
|
COPY ./src/open-r1-multimodal /workspace/src/open-r1-multimodal
|
|
|
|
# Install open_r1
|
|
WORKDIR /workspace/src/open-r1-multimodal
|
|
RUN pip install -e ".[dev]"
|
|
WORKDIR /workspace
|
|
|
|
# Install vllm
|
|
RUN pip install vllm==0.7.2
|
|
|
|
# Set environment variables for better Python output
|
|
ENV PYTHONUNBUFFERED=1
|
|
|
|
# Default command
|
|
CMD ["/bin/bash"]
|
|
|