# 使用PyTorch官方镜像作为基础
FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime

# 设置工作目录
WORKDIR /workspace

# 安装系统依赖
RUN apt-get update && apt-get install -y \
    build-essential \
    g++ \
    && rm -rf /var/lib/apt/lists/*

# 安装Python依赖
RUN pip install --no-cache-dir \
    gymnasium==0.28.1 \
    dm_control \
    shimmy[dm-control] \
    numpy==1.26.4 \
    pybind11

# 复制文件到容器
COPY gac_core.cpp setup.py ./
COPY eval_gac.py ./
COPY gac_pretrain_model/ ./gac_pretrain_model/

# 在容器内编译gac_core
RUN python setup.py build_ext --inplace

# 默认运行评估
CMD ["python", "eval_gac.py"]
