Add Dockerfile, docker-compose.yml, and Docker usage instructions

2026-07-15 00:49:48 +05:30
parent b4a77bbd2c
commit 44cf3839cd
3 changed files with 116 additions and 0 deletions
@@ -0,0 +1,47 @@
+# ── J-Wash Docker image ──────────────────────────────────────────────────────
+# Requires an NVIDIA GPU with CUDA 12.4+ drivers on the host.
+#   docker build -t j-wash .
+#   docker run --gpus all -p 8381:8381 -v ./data:/app/data -v ./hf_cache:/app/hf_cache j-wash
+
+FROM nvidia/cuda:12.4.1-runtime-ubuntu22.04 AS base
+
+SHELL ["/bin/bash", "-c"]
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    python3.11 python3.11-dev python3-pip python3-venv \
+    git curl ca-certificates \
+    && rm -rf /var/lib/apt/lists/*
+
+RUN curl -fsSL https://deb.nodesource.com/setup_20.x | bash - \
+    && apt-get install -y nodejs \
+    && rm -rf /var/lib/apt/lists/*
+
+RUN python3.11 -m venv /venv
+ENV PATH="/venv/bin:$PATH"
+
+WORKDIR /app
+
+COPY requirements.txt .
+
+RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cu124
+
+RUN git clone https://github.com/anthropics/jacobian-lens vendor/jacobian-lens \
+    && pip install -e vendor/jacobian-lens
+
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY ui/package.json ui/package-lock.json ui/
+RUN cd ui && npm ci
+
+COPY . .
+
+RUN cd ui && npm run build
+
+ENV HF_HOME=/app/hf_cache
+ENV JWASH_DATA_DIR=/app/data
+
+EXPOSE 8381
+
+VOLUME ["/app/data", "/app/hf_cache", "/app/lenses"]
+
+ENTRYPOINT ["python3.11", "-X", "utf8", "run.py"]