# 使用官方 PyTorch CPU 基础镜像
FROM pytorch/pytorch:latest

# 安装 Jupyter Lab 和常用库
RUN pip install --upgrade pip && \
    pip install jupyterlab matplotlib pandas seaborn

# 设置默认工作目录
WORKDIR /workspace

# 暴露 Jupyter 端口
EXPOSE 8888

# 默认启动 Jupyter Lab
CMD ["jupyter", "lab", "--ip=0.0.0.0", "--port=8888", "--no-browser", "--allow-root"]