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 -