Files
stack/notebooks/Dockerfile
kert 906d3aa318 Add GPU-accelerated notebook and Zotero desktop services
- notebooks: Marimo notebook server with CUDA/Polars GPU support
  - Uses nvidia/cuda base with uv package manager
  - Includes cudf-polars for GPU-accelerated dataframes
  - GPU benchmark comparing Polars CPU/GPU and Pandas

- zotero: Web-accessible Zotero via Selkies EGL desktop
  - Uses nvidia-egl-desktop with KasmVNC for browser access
  - GPU-accelerated desktop streaming
  - Persistent Zotero data directory

- compose.yml: Docker Compose orchestration for all services

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
2026-02-04 00:48:22 -05:00

48 lines
1.4 KiB
Docker

# syntax=docker/dockerfile:1
FROM nvidia/cuda:12.6.0-runtime-ubuntu24.04
ARG USERNAME=kert
ARG USER_UID=1000
ARG USER_GID=1000
ARG PYTHON_VERSION=3.13
ENV DEBIAN_FRONTEND=noninteractive \
PATH="/home/${USERNAME}/.local/bin:${PATH}" \
NVIDIA_VISIBLE_DEVICES=all \
NVIDIA_DRIVER_CAPABILITIES=compute,utility
# System dependencies
RUN apt-get update && apt-get install -y --no-install-recommends \
curl \
ca-certificates \
git \
build-essential \
&& rm -rf /var/lib/apt/lists/*
# Rename existing ubuntu user/group to kert and fix home ownership
RUN groupmod -n ${USERNAME} ubuntu \
&& usermod -l ${USERNAME} -d /home/${USERNAME} -m -s /bin/bash ubuntu \
&& chown -R ${USER_UID}:${USER_GID} /home/${USERNAME}
# Install uv
COPY --from=ghcr.io/astral-sh/uv:latest /uv /usr/local/bin/uv
COPY --from=ghcr.io/astral-sh/uv:latest /uvx /usr/local/bin/uvx
# Switch to user
USER ${USERNAME}
WORKDIR /home/${USERNAME}
# Initialize uv project and install dependencies
RUN uv python install ${PYTHON_VERSION} \
&& uv init workspace --python ${PYTHON_VERSION} \
&& cd workspace \
&& uv add "marimo[recommended]" polars cudf-polars-cu12 pandas numpy pyarrow \
--extra-index-url https://pypi.nvidia.com
EXPOSE 2718
CMD ["uv", "run", "--project", "/home/kert/workspace", \
"marimo", "edit", \
"--host", "0.0.0.0", "--port", "2718", "--headless", "--no-token", \
"/home/kert/notebooks"]