Validate RTX 3080 LeRobot environment

env/setup_lerobot.sh

@@ -18,7 +18,7 @@ esac
 [[ "$(uname -s)" == Linux && "$(uname -m)" == x86_64 ]] || {
   echo "Only Linux x86_64 GPU hosts are supported; RPi needs a separate environment." >&2; exit 1;
 }
-for cmd in uv nvidia-smi ffmpeg sha256sum; do
+for cmd in uv nvidia-smi ffmpeg sha256sum gcc; do
   command -v "$cmd" >/dev/null || { echo "Missing prerequisite: $cmd" >&2; exit 1; }
 done
 if ! $RESOLVE && [[ ! -s "$LOCK" ]]; then
@@ -34,12 +34,19 @@ exec > >(tee "$REPORT/setup.log") 2>&1
 echo "[1/5] Prerequisites"
 nvidia-smi --query-gpu=name,memory.total,driver_version --format=csv,noheader
 uv --version
+cat /etc/os-release > "$REPORT/os-release.txt"
+uname -srmo > "$REPORT/kernel.txt"
+nvidia-smi > "$REPORT/nvidia-smi.txt"
 ffmpeg -version > "$REPORT/ffmpeg.txt"
 # Set UV_DEFAULT_INDEX explicitly if a mirror is needed. No automatic Hub/token routing.
 echo "[2/5] Dedicated Python 3.12 environment"
 if [[ ! -d "$VENV" ]]; then uv venv --python 3.12 "$VENV"; fi
 PYTHON="${VENV}/bin/python"
 "$PYTHON" -c 'import sys; assert sys.version_info[:2] == (3, 12), sys.version'
+PYTHON_INCLUDE="$("$PYTHON" -c 'import sysconfig; print(sysconfig.get_path("include"))')"
+[[ -f "$PYTHON_INCLUDE/Python.h" ]] || {
+  echo "Missing Python development headers: $PYTHON_INCLUDE/Python.h" >&2; exit 1;
+}
 if $RESOLVE; then
   TEMP_LOCK="$(mktemp "${LOCK}.tmp.XXXXXX")"
   trap 'rm -f -- "$TEMP_LOCK"' EXIT