From bf169bf528493cd4076d366a780b4074c7694f51 Mon Sep 17 00:00:00 2001
From: Simon Alibert <alibert.sim@gmail.com>
Date: Wed, 22 May 2024 18:10:46 +0200
Subject: [PATCH] Fix paths

---
 docker/lerobot-gpu-dev/Dockerfile | 23 ++++++++++++++++-------
 1 file changed, 16 insertions(+), 7 deletions(-)

diff --git a/docker/lerobot-gpu-dev/Dockerfile b/docker/lerobot-gpu-dev/Dockerfile
index c36895f0..7e89df55 100644
--- a/docker/lerobot-gpu-dev/Dockerfile
+++ b/docker/lerobot-gpu-dev/Dockerfile
@@ -8,11 +8,11 @@ ARG DEBIAN_FRONTEND=noninteractive
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential cmake \
     git git-lfs openssh-client \
-    nano vim ffmpeg \
+    nano vim less util-linux \
     htop atop nvtop \
     sed gawk grep curl wget \
     tcpdump sysstat screen \
-    libglib2.0-0 libgl1-mesa-glx libegl1-mesa \
+    libglib2.0-0 libgl1-mesa-glx libegl1-mesa ffmpeg \
     python${PYTHON_VERSION} python${PYTHON_VERSION}-venv \
     && apt-get clean && rm -rf /var/lib/apt/lists/*
 
@@ -26,16 +26,25 @@ RUN (type -p wget >/dev/null || (apt update && apt-get install wget -y)) \
     && chmod go+r /etc/apt/keyrings/githubcli-archive-keyring.gpg \
     && echo "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/githubcli-archive-keyring.gpg] https://cli.github.com/packages stable main" | tee /etc/apt/sources.list.d/github-cli.list > /dev/null \
     && apt update \
-    && apt install gh -y
+    && apt install gh -y \
+    && apt clean && rm -rf /var/lib/apt/lists/*
 
 # Install miniconda
-RUN mkdir -p /miniconda3 \
-    && wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O /miniconda3/miniconda.sh \
-    && bash /miniconda3/miniconda.sh -b -u -p /miniconda3 \
-    && rm -rf /miniconda3/miniconda.sh
+RUN mkdir -p /root/miniconda3 \
+    && wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O /root/miniconda3/miniconda.sh \
+    && bash /root/miniconda3/miniconda.sh -b -u -p /root/miniconda3 \
+    && rm -rf /root/miniconda3/miniconda.sh
+ENV PATH="/root/miniconda3/bin/:$PATH"
+
+# Setup conda & create env
+RUN conda init bash \
+    && . ~/.bashrc \
+    && conda create -y -n lerobot python=3.10 \
+    && conda activate lerobot
 
 # Install poetry
 RUN curl -sSL https://install.python-poetry.org | python -
+ENV PATH="/root/.local/bin:$PATH"
 
 # Set EGL as the rendering backend for MuJoCo
 ENV MUJOCO_GL="egl"