File size: 1,203 Bytes
37a9361 dfc399f 37a9361 b9a4755 37a9361 89090d4 f7d5504 37a9361 f7d5504 5d14e50 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
# Configure image
ARG PYTHON_VERSION=3.10
FROM python:${PYTHON_VERSION}-slim
ARG PYTHON_VERSION
ARG DEBIAN_FRONTEND=noninteractive
# Install apt dependencies
RUN apt-get update && apt-get install -y --no-install-recommends \
build-essential cmake git wget \
libglib2.0-0 libgl1-mesa-glx libegl1-mesa ffmpeg \
&& apt-get clean && rm -rf /var/lib/apt/lists/*
# Create virtual environment
RUN ln -s /usr/bin/python${PYTHON_VERSION} /usr/bin/python
RUN python -m venv /opt/venv
ENV PATH="/opt/venv/bin:$PATH"
RUN echo "source /opt/venv/bin/activate" >> /root/.bashrc
RUN useradd -m -u 1000 user
# Install LeRobot
RUN git clone --branch user/rcadene/2024_05_17_improve_visualize_dataset https://github.com/huggingface/lerobot.git /lerobot
WORKDIR /lerobot
RUN pip install --upgrade --no-cache-dir pip
RUN pip install --no-cache-dir "." \
--extra-index-url https://download.pytorch.org/whl/cpu
RUN pip install --no-cache-dir flask
COPY --chown=user . /lerobot
CMD ["python", "lerobot/scripts/visualize_dataset_html.py", "--repo-id", "lerobot/koch_pick_place_lego_simple_v2", "--port", "7860", "--output-dir", "/tmp/outputs/visualize_dataset_html/lerobot/koch_pick_place_lego_simple_v2"]
|