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

# 安装 Jupyter Lab 和常用库
RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple && \
    pip install --upgrade pip && \
    pip install jupyterlab matplotlib pandas seaborn tensorboard visdom

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

# 暴露端口：Jupyter、TensorBoard、Visdom
EXPOSE 8888 6006 8097

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