stepankuzmin/pytorch-notebook

Jupyter Notebook PyTorch 镜像栈

镜像概述

本镜像基于 https://github.com/jupyter/docker-stacks/tree/master/scipy-notebook%EF%BC%8C%E9%9B%86%E6%88%90PyTorch%E6%A1%86%E6%9E%B6%EF%BC%8C%E6%8F%90%E4%BE%9B%E5%BC%80%E7%AE%B1%E5%8D%B3%E7%94%A8%E7%9A%84%E6%9C%BA%E5%99%A8%E5%AD%A6%E4%B9%A0%E4%B8%8E%E6%95%B0%E6%8D%AE%E5%88%86%E6%9E%90%E7%8E%AF%E5%A2%83%E3%80%82

核心功能

• Jupyter Notebook 4.3.x版本
• Conda Python 3.x和2.7.x双环境支持
• 预安装pandas、matplotlib、scipy、seaborn、scikit-learn、PyTorch等常用库
• 非特权用户jovyan（可配置UID），拥有/home/jovyan和/opt/conda目录所有权
• 内置tini入口点及start-notebook.sh、start-singleuser.sh等启动脚本
• 支持HTTPS、密码认证、无密码sudo等配置选项

使用场景

• 机器学习模型开发与训练
• 数据分析与可视化
• 教学演示与实验
• JupyterHub单用户实例部署

配置说明

基本使用

启动容器并监听8888端口（生成随机认证令牌）：

bash
docker run -it --rm -p 8888:8888 stepankuzmin/pytorch-notebook

访问时需使用日志中的认证令牌登录。

Notebook选项

传递Jupyter参数（例如设置自定义密码，需先通过IPython.lib.passwd()生成哈希）：

bash
docker run -d -p 8888:8888 stepankuzmin/pytorch-notebook start-notebook.sh --NotebookApp.password='sha1:74ba40f8a388:c913541b7ee99d15d5ed31d4226bf7838f83a50e'

Docker选项

• -e GEN_CERT=yes：生成自签名SSL证书启用HTTPS
• -e NB_UID=1000：指定jovyan用户UID（需以root运行）
• -e GRANT_SUDO=yes：授予无密码sudo权限（需以root运行）
• -v /host/work:/home/jovyan/work：挂载主机目录持久化数据

SSL证书配置

挂载自定义证书启用HTTPS：

bash
docker run -d -p 8888:8888 \
    -v /host/ssl:/etc/ssl/notebook \
    stepankuzmin/pytorch-notebook start-notebook.sh \
    --NotebookApp.keyfile=/etc/ssl/notebook/notebook.key \
    --NotebookApp.certfile=/etc/ssl/notebook/notebook.crt

Conda环境切换

切换到Python 2.7环境：

bash
source activate python2

返回默认Python 3环境：

bash
source deactivate

替代命令

• JupyterHub单用户实例：

  bash
  docker run -d stepankuzmin/pytorch-notebook /usr/local/bin/start-singleuser.sh
  

• 运行IPython控制台：

  bash
  docker run -it --rm stepankuzmin/pytorch-notebook start.sh ipython