# 使用官方的paddle镜像作为基础
FROM ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlex/paddlex:paddlex3.1.2-paddlepaddle3.0.0-gpu-cuda12.6-cudnn9.5-trt10.5

# 设置工作目录
WORKDIR /app

# 设置环境变量
ENV PYTHONUNBUFFERED=1 \
    # 设置时区
    TZ=Asia/Shanghai \
    # 设置pip镜像地址，加快安装速度
    PIP_INDEX_URL=https://pypi.tuna.tsinghua.edu.cn/simple

# 安装language-pack-en和openssh-server
RUN apt update && \
    apt install -y language-pack-en && \
    apt install -y openssh-server

# 配置SSH服务
RUN mkdir /var/run/sshd && \
    # 设置root密码，可根据需要修改
    echo 'root:fcb0102' | chpasswd && \
    # 允许root登录SSH
    sed -i 's/#PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/sshd_config

# 将当前目录内容复制到容器的/app内
COPY . /app

# 暴露22端口
EXPOSE 22

# 启动SSH服务
CMD ["/usr/sbin/sshd", "-D"]