17 lines
524 B
Docker
17 lines
524 B
Docker
# 使用官方 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"]
|