© 2026 Hedgehog Software, LLC
FROM runpod/pytorch:2.8.0-py3.11-cuda12.8.1-cudnn-devel-ubuntu22.04 RUN pip install ninja ENV MAX_JOBS=4 RUN MAX_JOBS=4 bash -c "ulimit -v 10485760 && pip install --no-build-isolation flash-attn"