# Example Dockerfile for torch executor.
# Base image version is an example — actual image configured via docker.images.
FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime
# Install everything the executor + fixtures need. bfloat16 fixtures need
# ml_dtypes; add per-op deps here.
RUN pip install numpy pyyaml ml_dtypes
COPY . /executor/
ENTRYPOINT ["python", "/executor/executor_main.py"]