98051ba0c9
- Move existing OndselSolver, GNN ML layer, and tooling into GNN/ directory for integration in later phases - Add Create addon scaffold: package.xml, Init.py - Add expression DAG with eval, symbolic diff, simplification - Add parameter table with fixed/free variable tracking - Add quaternion rotation as polynomial Expr trees - Add RigidBody entity (7 DOF: position + unit quaternion) - Add constraint classes: Coincident, DistancePointPoint, Fixed - Add Newton-Raphson solver with symbolic Jacobian + numpy lstsq - Add pre-solve passes: substitution + single-equation - Add DOF counting via Jacobian SVD rank - Add KindredSolver IKCSolver bridge for kcsolve integration - Add 82 unit tests covering all modules Registers as 'kindred' solver via kcsolve.register_solver() when loaded by Create's addon_loader.
62 lines
1.7 KiB
Docker
62 lines
1.7 KiB
Docker
FROM nvidia/cuda:12.4.1-devel-ubuntu22.04 AS base
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
ENV PYTHONUNBUFFERED=1
|
|
|
|
# System deps
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
python3.11 python3.11-venv python3.11-dev python3-pip \
|
|
git wget curl \
|
|
# FreeCAD headless deps
|
|
freecad \
|
|
libgl1-mesa-glx libglib2.0-0 \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.11 1
|
|
|
|
# Create venv
|
|
RUN python -m venv /opt/venv
|
|
ENV PATH="/opt/venv/bin:$PATH"
|
|
|
|
# Install PyTorch with CUDA
|
|
RUN pip install --no-cache-dir \
|
|
torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
|
|
|
|
# Install PyG
|
|
RUN pip install --no-cache-dir \
|
|
torch-geometric \
|
|
pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv \
|
|
-f https://data.pyg.org/whl/torch-2.4.0+cu124.html
|
|
|
|
WORKDIR /workspace
|
|
|
|
# Install project
|
|
COPY pyproject.toml .
|
|
RUN pip install --no-cache-dir -e ".[train,dev]" || true
|
|
|
|
COPY . .
|
|
RUN pip install --no-cache-dir -e ".[train,dev]"
|
|
|
|
# -------------------------------------------------------------------
|
|
FROM base AS cpu
|
|
|
|
# CPU-only variant (for CI and non-GPU environments)
|
|
FROM python:3.11-slim AS cpu-only
|
|
|
|
ENV PYTHONUNBUFFERED=1
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
git freecad libgl1-mesa-glx libglib2.0-0 \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
WORKDIR /workspace
|
|
|
|
COPY pyproject.toml .
|
|
RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
|
|
RUN pip install --no-cache-dir torch-geometric
|
|
|
|
COPY . .
|
|
RUN pip install --no-cache-dir -e ".[train,dev]"
|
|
|
|
CMD ["pytest", "tests/", "-v"]
|