jetware/pytorch-notebook

PyTorch Notebook镜像说明

镜像概述

Jetware/pytorch-notebook是预配置的最小化运行环境镜像，集成PyTorch机器学习库、Jupyter Notebook交互式编程工具与Python语言。镜像提供多硬件优化版本（x86 CPU的OpenBLAS/Intel MKL、NVIDIA GPU的CUDA）及多种操作系统支持，满足不同场景需求。

核心功能

1. 工具集成：包含PyTorch（机器学习）、Jupyter Notebook（浏览器端交互）、Python语言；
2. 硬件优化：支持CPU（OpenBLAS）、CPU_MKL（Intel加速）、CUDA9/CUDA10（GPU加速）；
3. 多OS适配：覆盖Alpine3.8、Ubuntu18.04、Debian9、CentOS7等系统；
4. 版本多样：支持PyTorch（1.0.0/0.4.1/0.3.1）与Python（3.6.3/2.7.14）版本组合。

使用场景

• 机器学习模型开发与训练；
• 数据科学分析与可视化；
• 编程教学与实验演示；
• 快速搭建临时开发环境。

配置说明

标签格式

镜像标签遵循规范：

pytorch-notebook:<pytorch_version>-python<python_version>-<optimization>-<operating_system>

示例：1.0.0-python3.6.3-cpu-alpine3.8（PyTorch1.0.0 + Python3.6.3 + CPU优化 + Alpine3.8）

常用标签

• latest：默认版本，对应1.0.0-python3.6.3-cpu-alpine3.8；
• latest-ubuntu18.04：Ubuntu18.04版本，对应1.0.0-python3.6.3-cpu-ubuntu18.04；
• latest-cpu_mkl：MKL加速版本，对应1.0.0-python3.6.3-cpu_mkl2018.2.199-alpine3.8；
• latest-python2：Python2版本，对应1.0.0-python2.7.14-cpu-alpine3.8。

部署示例

Docker Run命令

运行默认版本并映射端口：

bash
docker run -d -p 8888:8888 docker.xuanyuan.run/jetware/pytorch-notebook:latest

访问http://localhost:8888进入Jupyter界面。

Docker Compose示例

创建docker-compose.yml：

yaml
version: '3'
services:
  pytorch-notebook:
    image: docker.xuanyuan.run/jetware/pytorch-notebook:latest
    ports:
      - "8888:8888"
    volumes:
      - ./notebooks:/notebooks  # 本地目录挂载到容器
    restart: always

启动命令：

bash
docker-compose up -d