From 7bf584a3a9e0cb7448ff950e1c548ba21bab0fe4 Mon Sep 17 00:00:00 2001
From: Simon Alibert <alibert.sim@gmail.com>
Date: Wed, 22 May 2024 19:06:23 +0200
Subject: [PATCH] Revert "Remove apt-installed python"

This reverts commit ec96ba0bb0f0ee1eb85f339c388845bc2b4a3aae.
---
 docker/lerobot-gpu-dev/Dockerfile | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/docker/lerobot-gpu-dev/Dockerfile b/docker/lerobot-gpu-dev/Dockerfile
index d684e1be..59e0d6ff 100644
--- a/docker/lerobot-gpu-dev/Dockerfile
+++ b/docker/lerobot-gpu-dev/Dockerfile
@@ -13,8 +13,12 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     sed gawk grep curl wget \
     tcpdump sysstat screen \
     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/*
 
+# Setup `python`
+RUN ln -s /usr/bin/python${PYTHON_VERSION} /usr/bin/python
+
 # Install gh cli tool
 RUN (type -p wget >/dev/null || (apt update && apt-get install wget -y)) \
     && mkdir -p -m 755 /etc/apt/keyrings \
@@ -35,7 +39,8 @@ ENV PATH="/root/miniconda3/bin/:$PATH"
 # Setup conda & create env
 RUN conda init bash \
     && . ~/.bashrc \
-    && conda create -y -n lerobot python=${PYTHON_VERSION}
+    && conda create -y -n lerobot python=3.10 \
+    && conda activate lerobot
 
 # Install poetry
 RUN curl -sSL https://install.python-poetry.org | python -