FROM python:3.12.13-bookworm

# 2. Install system dependencies
RUN apt-get update && \
    apt-get install -y --no-install-recommends build-essential python3-dev rustc cargo && \
    rm -rf /var/lib/apt/lists/*

ENV PYTHONUNBUFFERED=1

WORKDIR /workspace

# Install Python dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir -U pip setuptools wheel && \
    pip install --no-cache-dir -r requirements.txt

# Build and install the rwkv Python package from source
COPY rwkv_pip_package rwkv_pip_package
RUN pip install --no-cache-dir -e rwkv_pip_package

# Copy the rest of the repository (optional for runtime)
COPY . .

# Default command: quick sanity test to ensure package imports
CMD ["python", "-c", "import rwkv, sys; print('rwkv version:', getattr(rwkv, '__version__', 'unknown'))"]

# Ignore git and various temporary files
.git/
__pycache__/
*.pyc
*.pyo
*.pyd
*.tmp
*.log
*.swp
*.bak
*.old
.DS_Store
/build/
/dist/
/venv/
/ENV/
/env/
/node_modules/
 rwkv_pip_package/__pycache__/
*.egg-info/
*.egg
*.whl
**/__pycache__/
rwkv_pip_package/build/
rwkv_pip_package/dist/
rwkv_pip_package/*.egg-info/

try:
    import rwkv
    ver = getattr(rwkv, '__version__', None)
    if ver:
        print("rwkv version:", ver)
    else:
        print("rwkv module loaded at:", getattr(rwkv, '__file__', 'unknown'))
except Exception as e:
    print("RWKV import failed:", e)
    raise

FROM python:3.12.13-bookworm

# 2. Install system dependencies
RUN apt-get update && \
    apt-get install -y --no-install-recommends build-essential python3-dev rustc cargo && \
    rm -rf /var/lib/apt/lists/*

ENV PYTHONUNBUFFERED=1

WORKDIR /workspace

# Install Python dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir -U pip setuptools wheel && \
    pip install --no-cache-dir -r requirements.txt

# Build and install the rwkv Python package from source
COPY rwkv_pip_package rwkv_pip_package
RUN pip install --no-cache-dir -e rwkv_pip_package

# Copy the rest of the repository (optional for runtime)
COPY . .

# Run a simple import test at startup to verify import
COPY rwkv_import_test.py rwkv_import_test.py

CMD ["python", "rwkv_import_test.py"]

FROM python:3.12.13-bookworm

ENV DEBIAN_FRONTEND=noninteractive

# Install essential system dependencies (avoid Rust for smaller image; rely on wheels)
RUN apt-get update && \
    apt-get install -y --no-install-recommends build-essential python3-dev && \
    rm -rf /var/lib/apt/lists/*

ENV PYTHONUNBUFFERED=1

WORKDIR /workspace

# Install Python dependencies (tokenizers may pull in prebuilt wheels when available)
COPY requirements.txt .
RUN python -m pip install --no-cache-dir -U pip setuptools wheel && \
    pip install --no-cache-dir -r requirements.txt

# Build and install the rwkv Python package from source (editable)
COPY rwkv_pip_package rwkv_pip_package
RUN pip install --no-cache-dir -e rwkv_pip_package

# Copy the rest of the repository (optional for runtime)
COPY . .

# Default command: a simple import test to verify the package is importable
CMD ["python", "-c", "import rwkv; print('rwkv ready:', getattr(rwkv, '__version__', 'unknown'))"]

FROM python:3.12.13-bookworm

ENV DEBIAN_FRONTEND=noninteractive

# Install essential system dependencies (avoid Rust for smaller image; rely on wheels)
RUN apt-get update && \
    apt-get install -y --no-install-recommends build-essential python3-dev && \
    rm -rf /var/lib/apt/lists/*

ENV PYTHONUNBUFFERED=1

WORKDIR /workspace

# Install Python dependencies (tokenizers may pull in prebuilt wheels when available)
COPY requirements.txt .
RUN python -m pip install --no-cache-dir -U pip setuptools wheel && \
    pip install --no-cache-dir -r requirements.txt

# Build and install the rwkv Python package from source (editable)
COPY rwkv_pip_package rwkv_pip_package
RUN pip install --no-cache-dir -e rwkv_pip_package

# Copy the rest of the repository (optional for runtime)
COPY . .

# Default command: a simple import test to verify the package is importable
COPY rwkv_import_test.py rwkv_import_test.py

CMD ["python", "rwkv_import_test.py"]

FROM python:3.12.13-bookworm

ENV DEBIAN_FRONTEND=noninteractive

# Install essential system dependencies (avoid Rust for smaller image; rely on wheels)
RUN apt-get update && \
    apt-get install -y --no-install-recommends build-essential python3-dev rustc cargo && \
    rm -rf /var/lib/apt/lists/*

ENV PYTHONUNBUFFERED=1

WORKDIR /workspace

# Install Python dependencies (tokenizers may pull in prebuilt wheels when available)
COPY requirements.txt .
RUN python -m pip install --no-cache-dir -U pip setuptools wheel && \
    pip install --no-cache-dir -r requirements.txt

# Build and install the rwkv Python package from source (editable)
COPY rwkv_pip_package rwkv_pip_package
RUN pip install --no-cache-dir -e rwkv_pip_package

# Copy the rest of the repository (optional for runtime)
COPY . .

# Copy a small check script to verify importability
COPY check_rwkv.py check_rwkv.py

# Default command: run the check script to verify that rwkv imports
CMD ["python", "check_rwkv.py"]

try:
    import rwkv
    print('RWKV import OK')
    print('version:', getattr(rwkv, '__version__', 'unknown'))
except Exception as e:
    print('RWKV import FAILED:', e)
    raise

# Ignore git and various temporary files
.git/
__pycache__/
*.pyc
*.pyo
*.pyd
*.tmp
*.log
*.swp
*.bak
*.old
.DS_Store
/build/
/dist/
/venv/
/ENV/
/env/
/node_modules/
rwkv_pip_package/__pycache__/
*.egg-info/
*.egg
*.whl
**/__pycache__/
rwkv_pip_package/build/
rwkv_pip_package/dist/
rwkv_pip_package/*.egg-info/

FROM python:3.12.13-bookworm

ENV DEBIAN_FRONTEND=noninteractive

# Install essential system dependencies (allow Rust for building tokenizers)
RUN apt-get update && \
    apt-get install -y --no-install-recommends build-essential python3-dev rustc cargo && \
    rm -rf /var/lib/apt/lists/*

ENV PYTHONUNBUFFERED=1

WORKDIR /workspace

# Install Python dependencies (tokenizers may pull in prebuilt wheels when available)
COPY requirements.txt .
RUN python -m pip install --no-cache-dir -U pip setuptools wheel && \
    pip install --no-cache-dir -r requirements.txt

# Build and install the rwkv Python package from source (editable)
COPY rwkv_pip_package rwkv_pip_package
RUN pip install --no-cache-dir -e rwkv_pip_package

# Copy the rest of the repository (runtime)
COPY . .

# Copy a small check script to verify importability
COPY check_rwkv.py check_rwkv.py

# Default command: run the check script to verify the package is importable
CMD ["python", "check_rwkv.py"]

FROM python:3.12.13-bookworm

ENV DEBIAN_FRONTEND=noninteractive

# Install essential system dependencies (anticipate building tokenizers/pip dependencies)
RUN apt-get update && \
    apt-get install -y --no-install-recommends build-essential python3-dev rustc cargo && \
    rm -rf /var/lib/apt/lists/*

ENV PYTHONUNBUFFERED=1

WORKDIR /workspace

# Install Python dependencies (tokenizers may need Rust for building wheels)
COPY requirements.txt .
RUN python -m pip install --no-cache-dir -U pip setuptools wheel && \
    pip install --no-cache-dir -r requirements.txt

# Build and install the rwkv Python package from source (editable)
COPY rwkv_pip_package rwkv_pip_package
RUN pip install --no-cache-dir -e rwkv_pip_package

# Copy a lightweight smoke test to verify importability at runtime
COPY smoke_test.py smoke_test.py

CMD ["python", "smoke_test.py"]

try:
    import rwkv
    print('RWKV import OK')
    print('version:', getattr(rwkv, '__version__', 'unknown'))
except Exception as e:
    print('RWKV import FAILED:', e)

FROM python:3.12.13-bookworm

# Environment
ENV LANG=C.UTF-8
ENV LC_ALL=C.UTF-8
ENV PYTHONUNBUFFERED=1

# Install system dependencies required for building Python extensions (Rust, C/C++)
RUN apt-get update \
    && apt-get install -y --no-install-recommends \
        build-essential \
        python3-dev \
        rustc \
        cargo \
    && rm -rf /var/lib/apt/lists/*

# Working directory
WORKDIR /workspace

# Install Python dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# Install PyTorch CPU version (no CUDA)
RUN pip install --no-cache-dir --upgrade pip setuptools wheel \
    && pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

# Copy the package source (avoid shadowing installed package)
COPY rwkv_pip_package rwkv_pip_package

# Build and install the RWKV package from source
RUN pip install --no-cache-dir -e rwkv_pip_package

# Remove build tools to reduce image size
RUN apt-get purge -y --auto-remove build-essential python3-dev rustc cargo \
    && rm -rf /var/lib/apt/lists/*

# Optional: expose a minimal command to verify the installation.
# Try to import the package and print its version.
CMD ["python", "-c", "import rwkv, sys; print('RWKV package version:', getattr(rwkv, '__version__', 'unknown')); print('Python', sys.version)" ]

# Exclude VCS and build artifacts
.git
.github/
**/__pycache__/
**/*.pyc
**/*.pyo
node_modules/
build/
dist/
*.egg-info/
*.log
*.tmp
rwkv_pip_package/.git/
rwkv_pip_package/build/
rwkv_pip_package/dist/
README.md
LICENSE

# Exclude VCS and build artifacts
.git
.github/
**/__pycache__/ 
**/*.pyc
**/*.pyo
node_modules/
build/
dist/
*.egg-info/
*.log
*.tmp
rwkv_pip_package/.git/
rwkv_pip_package/build/
rwkv_pip_package/dist/
README.md
LICENSE

# Exclude VCS and build artifacts
.git
.github/
**/__pycache__/
**/*.pyc
**/*.pyo
node_modules/
build/
dist/
*.egg-info/
*.log
*.tmp
rwkv_pip_package/.git/
rwkv_pip_package/build/
rwkv_pip_package/dist/
README.md
LICENSE

try:
    import rwkv
    ver = getattr(rwkv, '__version__', 'unknown')
except Exception as e:
    ver = f'error: {e}'
import sys
print('RWKV smoke test')
print('RWKV version:', ver)
print('Python:', sys.version.split()[0])

FROM python:3.12.13-bookworm

# Environment
ENV LANG=C.UTF-8
ENV LC_ALL=C.UTF-8
ENV PYTHONUNBUFFERED=1

# Install system dependencies required for building Python extensions (Rust, C/C++)
RUN apt-get update \
    && apt-get install -y --no-install-recommends \
        build-essential \
        python3-dev \
        rustc \
        cargo \
    && rm -rf /var/lib/apt/lists/*

# Working directory
WORKDIR /workspace

# Install Python dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# Install PyTorch CPU version (no CUDA)
RUN pip install --no-cache-dir --upgrade pip setuptools wheel \
    && pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

# Copy the package source (avoid shadowing installed package)
COPY rwkv_pip_package rwkv_pip_package

# Build and install the RWKV package from source
RUN pip install --no-cache-dir -e rwkv_pip_package

# Remove build tools to reduce image size
RUN apt-get purge -y --auto-remove build-essential python3-dev rustc cargo \
    && rm -rf /var/lib/apt/lists/*

# Copy smoke test
COPY smoke_test.py ./smoke_test.py

# Optional: expose a minimal command to verify the installation.
# Try to import the package and print its version.
CMD ["python", "smoke_test.py"]

FROM python:3.12.13-bookworm

# Environment
ENV LANG=C.UTF-8
ENV LC_ALL=C.UTF-8
ENV PYTHONUNBUFFERED=1

# Install system dependencies required for building Python extensions (Rust, C/C++)
RUN apt-get update \
    && apt-get install -y --no-install-recommends \
        build-essential \
        python3-dev \
        rustc \
        cargo \
    && rm -rf /var/lib/apt/lists/*

# Working directory
WORKDIR /workspace

# Install Python dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# Install PyTorch CPU version (no CUDA)
RUN pip install --no-cache-dir --upgrade pip setuptools wheel \
    && pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

# Prepare Python path for local RWKV source
RUN mkdir -p /workspace/src
# Copy the RWKV package source from the local repo into the image so it can be imported as `rwkv`
COPY rwkv_pip_package/src /workspace/src/rwkv
ENV PYTHONPATH=/workspace/src:${PYTHONPATH}

# Remove build tools to reduce image size
RUN apt-get purge -y --auto-remove build-essential python3-dev rustc cargo \
    && rm -rf /var/lib/apt/lists/*

# Copy smoke test
COPY smoke_test.py ./smoke_test.py

# Optional: expose a minimal command to verify the installation.
# Try to import the package and print its version.
CMD ["python", "smoke_test.py"]

FROM python:3.12.13-bookworm

# Basic environment settings
ENV LANG=C.UTF-8
ENV LC_ALL=C.UTF-8
ENV PYTHONUNBUFFERED=1

# System dependencies for building Python extensions
RUN apt-get update \
    && apt-get install -y --no-install-recommends \
        build-essential \
        python3-dev \
        rustc \
        cargo \
    && rm -rf /var/lib/apt/lists/*

WORKDIR /workspace

# Install Python runtime requirements
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# Install the RWKV package in editable mode from source
COPY rwkv_pip_package rwkv_pip_package
RUN pip install --no-cache-dir -e rwkv_pip_package

# Copy smoke test script to verify build
COPY smoke_test.py ./smoke_test.py

# Default command runs a quick smoke test
CMD ["python", "smoke_test.py"]

Concise actionable summary of the RWKV tool output

What it is
- The RWKV Language Model loader/runner with multiple backends (CUDA, DirectML, CPU fallback) and versioned variants (v4, v5.x, v6.0, v7 Goose).

Key environment flags and effects
- RWKV_JIT_ON: if not '0', uses JIT ScriptModule/script_method; otherwise normal torch.nn.Module.
- RWKV_CUDA_ON: if '1', builds CUDA extensions (wrapper.cpp, operators.cu, gemm_fp16_cublas.cpp); on failure, falls back to a smaller build (wrapper.cpp, operators.cu) and sets DISABLE_CUBLAS_GEMM=True.
- RWKV_DML_ON: enables DirectML support.
- RWKV_V7_ON: enables RWKV-7 Goose variant (loads wkv7 kernels, sets HEAD_SIZE=64).
- RWKV_DE_VERSION: if '1', enables DeepEmbed version and merges embeddings when present.
- RWKV_RESCALE_LAYER: controls layer-rescaling (fp16 overflow mitigation). Default 6 for fp16 strategy; 0 otherwise.

Important files/paths
- current_path = directory of this script (os.path.dirname(os.path.abspath(__file__)))
- CUDA sources if building: cuda/wrapper.cpp, cuda/operators.cu, cuda/gemm_fp16_cublas.cpp; optional: cuda/rwkv7_op.cpp, cuda/rwkv7.cu (for v7 Goose).
- Model file expected: MODEL_NAME.pth (or supplied MODEL_NAME ending with .pth)
- Conversion/strategy metadata stored in model dict keys (e.g., _strategy, _version, _rescale_layer)

Versioning and capabilities
- Model versioning detected from weights:
  - v4 baseline; possible v5.x, v6.0, v5.2, v6.0 depending on keys (time_decay, time_first, time_maa, time_faaaa, etc.)
- When RWKV_V7_ON is '1', RWKV becomes RWKV_x070 path with Goose-style operators and options.

Model loading and strategy
- Loads model (.pth) to CPU first, then moves to target devices per strategy.
- Computes per-layer strategy, allocating layers across devices, and prints a Strategy summary:
  - “Strategy: (total L+1 layers)”
  - Per-layer lines showing device, data type, and storage/streaming info.
- If ALREADY_CONVERTED is present in the model, it validates strategy/version and uses pre-converted weights.

Runtime behavior and kernels
- CUDA path defines:
  - cuda_wkv (forward for RWKV), cuda_mm8_seq/one for 8-bit weight math.
  - mm8/xmm wrappers and a matmul_float path using cuBLAS when available.
- Fallbacks to tensor ops (torch.matmul) when CUDA kernels are disabled or unavailable.
- Forward paths differ by version (attn/ffn implementations) and by seq vs single-token mode.

Notable messages and errors
- “Failed to build cuBLAS matmul, falling back to torch.matmul. Small model with fp16 will overflow.” (action: indicates fallback and potential FP16 overflow risk)

Usage hints (actionable)
- To enable CUDA kernels: set RWKV_CUDA_ON=1 and ensure a CUDA-capable PyTorch build.
- To use RWKV-7 Goose: set RWKV_V7_ON=1 and ensure related CUDA files are present.
- To avoid JIT or force non-JIT: set RWKV_JIT_ON accordingly.
- For DeepEmbed, set RWKV_DE_VERSION=1 and inspect merged embedding behavior.
- Model name usage: provide MODEL_NAME.pth; code will normalize by appending .pth if missing.

This preserves actionable items: environment flags, paths, version behavior, conversion steps, and the known CUDA build error.

# Smoke test shim for import path rwkv_pip_package
__version__ = "0.0.0"
try:
    # Attempt to import the real package if available
    from rwkv import *  # noqa: F401,F403
except Exception:
    pass

Placeholder

FROM python:3.12.13-bookworm

ENV LANG=C.UTF-8

# Work in a dedicated workspace
WORKDIR /workspace

# Basic tooling
# Upgrade packaging tools first
RUN python -m pip install --upgrade pip setuptools wheel

# Install system dependencies
RUN apt-get update && \
    apt-get install -y --no-install-recommends build-essential python3-dev rustc cargo && \
    rm -rf /var/lib/apt/lists/*

# Copy project source
COPY . .

# Install Python dependencies
RUN pip install --no-cache-dir --upgrade pip
RUN pip install --no-cache-dir -r requirements.txt
RUN pip install --no-cache-dir -e rwkv_pip_package
# Install PyTorch CPU variant (no CUDA) for runtime
RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

# Basic smoke test to verify importability of the package
RUN python - <<'PY'
import sys, importlib
for m in ["rwkv", "rwkv_pip_package"]:
  try:
    mod = importlib.import_module(m)
    print("IMPORT_OK:", m, getattr(mod, '__version__', 'N/A'))
  except Exception as e:
    print("IMPORT_FAIL:", m, e)
PY

# Entry point wrapper to group quick checks and allow interactive use
RUN printf '%s
' "#!/bin/sh" "set -e" "python - <<'PY'" "import sys, importlib" "for m in ['rwkv','rwkv_pip_package']:" "  try:" "    mod = importlib.import_module(m)" "    print('IMPORT_OK:', m, getattr(mod, '__version__', 'N/A'))" "  except Exception as e:" "    print('IMPORT_FAIL:', m, e)" "PY" "> /entrypoint.sh && chmod +x /entrypoint.sh

# Default to an interactive shell if no command is provided
CMD ["/bin/bash"]

git
__pycache__
*.pyc
*.pyo
build
dist
*.egg-info
*.log
.pytest_cache

#0 building with "default" instance using docker driver

#1 [internal] load build definition from Dockerfile
#1 transferring dockerfile: 1.59kB done
#1 DONE 0.0s
Dockerfile:40
--------------------
  38 |     # Entry point wrapper to group quick checks and allow interactive use
  39 |     RUN printf '%s
  40 | >>> ' "#!/bin/sh" "set -e" "python - <<'PY'" "import sys, importlib" "for m in ['rwkv','rwkv_pip_package']:" "  try:" "    mod = importlib.import_module(m)" "    print('IMPORT_OK:', m, getattr(mod, '__version__', 'N/A'))" "  except Exception as e:" "    print('IMPORT_FAIL:', m, e)" "PY" "> /entrypoint.sh && chmod +x /entrypoint.sh
  41 |     
  42 |     # Default to an interactive shell if no command is provided
--------------------
ERROR: failed to build: failed to solve: dockerfile parse error on line 40: unknown instruction: '

FROM python:3.12.13-bookworm

ENV LANG=C.UTF-8

WORKDIR /workspace

# Basic tooling
# Upgrade packaging tools first
RUN python -m pip install --upgrade pip setuptools wheel

# Install system dependencies
RUN apt-get update && \
    apt-get install -y --no-install-recommends build-essential python3-dev rustc cargo && \
    rm -rf /var/lib/apt/lists/*

# Copy project source
COPY . .

# Install Python dependencies
RUN pip install --no-cache-dir -r requirements.txt
RUN pip install --no-cache-dir -e rwkv_pip_package
# Install PyTorch CPU variant (no CUDA) for runtime
RUN python -m pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

# Basic smoke test to verify importability of the package
RUN python - <<'PY'
import sys
try:
    import rwkv
    print("IMPORT_OK:\nRWKV version:", getattr(rwkv, '__version__', 'N/A'))
except Exception as e:
    print("IMPORT_FAIL:", e)
PY

# Default to an interactive shell
CMD ["/bin/bash"]

git
__pycache__
*.pyc
*.pyo
build/
dist/
*.egg-info/
*.log
.pytest_cache/

__version__ = '0.0.0'

- Exact error message and exit code:
  - Exit code: not shown in the log
  - Error messages:
    - ERROR: failed to prepare extraction snapshot "extract-18202245-DN4X sha256:363c80c87d386a7e3890725a4c112d3948fdde190d9eb23f3febb13a84538564": parent snapshot sha256:f922c784e78dcd216dd704282b1d365e6b066fe8a54d0958dd556cb582ab7c1c does not exist: not found
    - > exporting to image:
    - ERROR: failed to build: failed to solve: failed to prepare extraction snapshot "extract-18202245-DN4X sha256:363c80c87d386a7e3890725a4c112d3948fdde190d9eb23f3febb13a84538564": parent snapshot sha256:f922c784e78dcd216dd704282b1d365e6b066fe8a54d0958dd556cb582ab7c1c does not exist: not found

- Failing command/step:
  - Step: exporting to image (the final build/export phase, after 9 steps)
  - Root cause shown: failure during extraction snapshot preparation; parent snapshot does not exist

- Missing packages or files mentioned:
  - None explicitly mentioned as missing. The error is in the image export/extraction snapshot phase, not a missing package/file error.

- Version mismatch info:
  - No explicit version mismatch errors observed.
  - Base image used: python:3.12.13-bookworm
  - Torch stack downloaded: torch 2.10.0+cpu, torchvision 0.25.0+cpu, torchaudio 2.10.0+cpu
  - RWKV check printed: RWKV version: 0.0.0 (not an error; just a version reported)

__version__ = '0.0.1'

FROM python:3.12.13-bookworm

ENV LANG=C.UTF-8

WORKDIR /workspace

# Basic tooling
# Upgrade packaging tools first
RUN python -m pip install --upgrade pip setuptools wheel

# Install system dependencies
RUN apt-get update && \
    apt-get install -y --no-install-recommends build-essential python3-dev rustc cargo && \
    rm -rf /var/lib/apt/lists/*

# Copy project source
COPY . .

# Install Python dependencies
RUN pip install --no-cache-dir -r requirements.txt
RUN pip install --no-cache-dir -e rwkv_pip_package
# Install PyTorch CPU variant (no CUDA) for runtime
RUN python -m pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

# Basic smoke test to verify importability of the package
RUN python - <<'PY'
import sys
ok = True
try:
    import rwkv
    print("IMPORT_OK RWKV:", getattr(rwkv, '__version__', 'N/A'))
except Exception as e:
    print("IMPORT_FAIL RWKV:", e)
    ok = False
try:
    import torch
    print("TORCH_OK:", torch.__version__, "CUDA_AVAILABLE=", torch.cuda.is_available())
except Exception as e:
    print("TORCH_FAIL:", e)
    ok = False
sys.exit(0 if ok else 1)
PY

# Default to an interactive shell
CMD ["/bin/bash"]

FROM python:3.12.13-bookworm

ENV LANG=C.UTF-8

# Create a non-root user for better security during runtime
RUN groupadd -r app && useradd -m -r -g app app

WORKDIR /workspace

# Basic tooling
RUN python -m pip install --upgrade pip setuptools wheel

# Install system dependencies
RUN apt-get update && \
    apt-get install -y --no-install-recommends build-essential python3-dev rustc cargo && \
    rm -rf /var/lib/apt/lists/*

# Copy project source
COPY . .

# Install Python dependencies
RUN pip install --no-cache-dir -r requirements.txt
RUN pip install --no-cache-dir -e rwkv_pip_package

# Install PyTorch CPU variant (no CUDA) for runtime
RUN python -m pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

# Ensure proper permissions for the non-root user
RUN chown -R app:app /workspace
USER app

# Default to an interactive shell (can be overridden by docker run command)
CMD ["/bin/bash"]