Dockerfile for examples

.dockerignore

@@ -0,0 +1,29 @@
+# Ignore pytest cache
+.pytest_cache
+
+# Ignore runs directory
+runs
+
+# Ignore videos directory
+videos
+
+# Ignore wandb directory
+wandb
+
+# Python cache files
+__pycache__
+*.pyc
+*.pyo
+*.pyd
+
+# Virtual environment
+venv
+.env
+
+# IDE specific files
+.vscode
+.idea
+
+# OS generated files
+.DS_Store
+Thumbs.db

Dockerfile

@@ -0,0 +1,63 @@
+# Start with the latest NVIDIA CUDA base image
+FROM nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04
+
+# Set environment variables
+ENV PYTHONUNBUFFERED=1 \
+    PYTHONDONTWRITEBYTECODE=1 \
+    PIP_NO_CACHE_DIR=off \
+    PIP_DISABLE_PIP_VERSION_CHECK=on \
+    PIP_DEFAULT_TIMEOUT=100 \
+    POETRY_VERSION=1.6.1 \
+    POETRY_HOME="/opt/poetry" \
+    POETRY_VIRTUALENVS_IN_PROJECT=true \
+    POETRY_NO_INTERACTION=1 \
+    PYSETUP_PATH="/opt/pysetup" \
+    VENV_PATH="/opt/pysetup/.venv"
+
+ENV DEBIAN_FRONTEND=noninteractive
+ENV PYTHONUNBUFFERED 1
+ENV PIP_NO_CACHE_DIR 1
+ENV PIP_ROOT_USER_ACTION=ignore
+RUN apt-get update && apt-get install -y \
+  ffmpeg git vim curl software-properties-common \
+  libglew-dev x11-xserver-utils xvfb
+
+
+# Set work directory
+WORKDIR $PYSETUP_PATH
+
+# Install system dependencies
+RUN apt-get update \
+    && apt-get install -y --no-install-recommends \
+        curl \
+        build-essential \
+        python3.11 \
+        python3.11-dev \
+        python3.11-venv \
+        python3-pip \
+    && rm -rf /var/lib/apt/lists/* \
+    && apt-get clean
+
+# Install Poetry
+RUN curl -sSL https://install.python-poetry.org | python3 -
+
+# Add Poetry to PATH
+ENV PATH="$POETRY_HOME/bin:$VENV_PATH/bin:$PATH"
+
+# Copy .dockerignore
+COPY .dockerignore .
+
+# Copy poetry.lock* in case it doesn't exist in the repo
+COPY ./pyproject.toml ./poetry.lock* ./
+
+# Install dependencies
+RUN poetry install --no-root --no-dev --with examples
+
+# Copy project files
+COPY . .
+
+# Install project
+RUN poetry install --with examples
+
+# Make examples executable
+RUN chmod -R +x examples