2024-04-25 20:58:39 +08:00
|
|
|
FROM nvidia/cuda:12.4.1-base-ubuntu22.04
|
|
|
|
|
|
2025-02-26 01:42:35 +08:00
|
|
|
# Configure environment variables
|
2024-04-25 20:58:39 +08:00
|
|
|
ARG PYTHON_VERSION=3.10
|
2025-02-26 01:42:35 +08:00
|
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
|
|
ENV MUJOCO_GL="egl"
|
|
|
|
|
ENV PATH="/opt/venv/bin:$PATH"
|
2024-05-23 20:39:14 +08:00
|
|
|
|
2025-02-26 01:42:35 +08:00
|
|
|
# Install dependencies and set up Python in a single layer
|
2024-04-25 20:58:39 +08:00
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
2025-02-02 20:59:33 +08:00
|
|
|
build-essential cmake git git-lfs \
|
2024-06-15 00:11:19 +08:00
|
|
|
libglib2.0-0 libgl1-mesa-glx libegl1-mesa ffmpeg \
|
2025-02-01 19:06:11 +08:00
|
|
|
speech-dispatcher libgeos-dev \
|
2024-07-15 23:43:10 +08:00
|
|
|
python${PYTHON_VERSION}-dev python${PYTHON_VERSION}-venv \
|
2025-02-26 01:42:35 +08:00
|
|
|
&& ln -s /usr/bin/python${PYTHON_VERSION} /usr/bin/python \
|
|
|
|
|
&& python -m venv /opt/venv \
|
|
|
|
|
&& apt-get clean && rm -rf /var/lib/apt/lists/* \
|
|
|
|
|
&& echo "source /opt/venv/bin/activate" >> /root/.bashrc
|
2024-05-23 20:39:14 +08:00
|
|
|
|
2025-02-26 01:42:35 +08:00
|
|
|
# Clone repository and install LeRobot in a single layer
|
2024-04-25 20:58:39 +08:00
|
|
|
WORKDIR /lerobot
|
2025-02-26 01:42:35 +08:00
|
|
|
RUN git lfs install \
|
|
|
|
|
&& git clone https://github.com/huggingface/lerobot.git . \
|
|
|
|
|
&& /opt/venv/bin/pip install --upgrade --no-cache-dir pip \
|
|
|
|
|
&& /opt/venv/bin/pip install --no-cache-dir ".[test, aloha, xarm, pusht, dynamixel]"
|
